{
  "cells": [
    {
      "cell_type": "code",
      "execution_count": 2,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "6OK_F7lqYy6D",
        "outputId": "e04e4683-eb08-4181-b361-a89f95254831"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "/content/drive/.shortcut-targets-by-id/1-CXP81xST-t3P2hLZQLjfipgHlqh7k86/yolo-distillation\n"
          ]
        }
      ],
      "source": [
        "%cd /content/drive/MyDrive/yolo-distillation"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "GoEY18zpZSPq"
      },
      "outputs": [],
      "source": [
        "!pip install -r requirements.txt"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "from google.colab import drive\n",
        "drive.mount('/content/drive')"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "-ve_YCOfo80U",
        "outputId": "ac5a7565-3c2f-4354-e489-d755439a2230"
      },
      "execution_count": 1,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Mounted at /content/drive\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "GpArQA7mZZ3j",
        "outputId": "ec62cb04-5626-4682-f435-1acc62f5dbe5"
      },
      "outputs": [
        {
          "name": "stdout",
          "output_type": "stream",
          "text": [
            "/usr/local/lib/python3.7/dist-packages/torch/nn/modules\n"
          ]
        }
      ],
      "source": [
        "%cd /usr/local/lib/python3.7/dist-packages/torch/nn/modules"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 17
        },
        "id": "HiAy09cwZdz1",
        "outputId": "7add8f6a-5452-47e1-f7f9-af39ea559dbf"
      },
      "outputs": [
        {
          "data": {
            "application/javascript": [
              "\n",
              "      ((filepath) => {{\n",
              "        if (!google.colab.kernel.accessAllowed) {{\n",
              "          return;\n",
              "        }}\n",
              "        google.colab.files.view(filepath);\n",
              "      }})(\"/usr/local/lib/python3.7/dist-packages/torch/nn/modules/upsampling.py\")"
            ],
            "text/plain": [
              "<IPython.core.display.Javascript object>"
            ]
          },
          "metadata": {},
          "output_type": "display_data"
        }
      ],
      "source": [
        "from google.colab import files\n",
        "files.view('upsampling.py')"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "kINGxTCsZm9a",
        "outputId": "ac02132b-5e23-41d7-ac92-0a3282bdfa95"
      },
      "outputs": [
        {
          "name": "stdout",
          "output_type": "stream",
          "text": [
            "/content/drive/MyDrive/yolo-distillation\n"
          ]
        }
      ],
      "source": [
        "%cd /content/drive/MyDrive/yolo-distillation"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "8Y1CNJh5ZuZe"
      },
      "outputs": [],
      "source": [
        "!python val.py --data data/VOC_animal.yaml --weights weights/yolov5l_best.pt --batch-size 16"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "4xHwlOIfHNgy"
      },
      "outputs": [],
      "source": [
        "!python train_dist_v1.py --weights weights/yolov5n_best.pt --cfg models/yolov5n_animal.yaml --data data/VOC_animal.yaml --batch-size 16 --epochs 2 --workers 2 --t_weights weights/yolov5l_best.pt --hyp data/hyps/hyp.scratch-low-distillation.yaml --distill_cfg configs/distill_config.py --name yolov5n-distilled"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "G-319IyNfprQ"
      },
      "outputs": [],
      "source": [
        "!python train_dist_v2.py --weights weights/yolov5n_best.pt --cfg models/yolov5n_animal.yaml --data data/VOC_animal.yaml --batch-size 16 --epochs 2 --workers 2 --t_weights weights/yolov5l_best.pt --hyp data/hyps/hyp.scratch-low-distillation.yaml --distill_cfg configs/distill_config2.py --name yolov5n-distilled"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "oiBvFj5HZkAl",
        "outputId": "25adf6d4-5656-4218-a332-7734849f341b"
      },
      "outputs": [
        {
          "name": "stdout",
          "output_type": "stream",
          "text": [
            "Generating grammar tables from /usr/lib/python3.7/lib2to3/Grammar.txt\n",
            "Generating grammar tables from /usr/lib/python3.7/lib2to3/PatternGrammar.txt\n",
            "\u001b[34m\u001b[1mtrain_dist: \u001b[0mweights=weights/yolov5n_best.pt, t_weights=weights/yolov5l_best.pt, dist_loss=l2, temperature=20, distill_cfg=configs/distill_config2.py, cfg=models/yolov5n_animal.yaml, data=data/VOC_animal.yaml, hyp=data/hyps/hyp.scratch-low-distillation.yaml, epochs=50, batch_size=16, imgsz=640, rect=False, resume=False, nosave=False, noval=False, noautoanchor=False, evolve=None, bucket=, cache=None, image_weights=False, device=, multi_scale=False, single_cls=False, optimizer=SGD, sync_bn=False, workers=2, project=runs/train, name=yolov5n-distilled, exist_ok=False, quad=False, cos_lr=False, label_smoothing=0.0, patience=100, freeze=[0], save_period=-1, local_rank=-1, entity=None, upload_dataset=False, bbox_interval=-1, artifact_alias=latest\n",
            "Command 'git fetch && git config --get remote.origin.url' timed out after 5 seconds\n",
            "YOLOv5 🚀 3f3185b torch 1.12.1+cu113 CUDA:0 (Tesla T4, 15110MiB)\n",
            "\n",
            "\u001b[34m\u001b[1mhyperparameters: \u001b[0mlr0=0.01, lrf=0.01, momentum=0.937, weight_decay=0.0005, warmup_epochs=3.0, warmup_momentum=0.8, warmup_bias_lr=0.1, box=0.05, cls=0.5, cls_pw=1.0, obj=1.0, obj_pw=1.0, iou_t=0.2, anchor_t=4.0, fl_gamma=0.0, hsv_h=0.015, hsv_s=0.7, hsv_v=0.4, degrees=0.0, translate=0.1, scale=0.5, shear=0.0, perspective=0.0, flipud=0.0, fliplr=0.5, mosaic=1.0, mixup=0.0, copy_paste=0.0, dist=0.5\n",
            "\u001b[34m\u001b[1mWeights & Biases: \u001b[0mrun 'pip install wandb' to automatically track and visualize YOLOv5 🚀 runs (RECOMMENDED)\n",
            "\u001b[34m\u001b[1mTensorBoard: \u001b[0mStart with 'tensorboard --logdir runs/train', view at http://localhost:6006/\n",
            "\n",
            "                 from  n    params  module                                  arguments                     \n",
            "  0                -1  1      1760  models.common.Conv                      [3, 16, 6, 2, 2]              \n",
            "  1                -1  1      4672  models.common.Conv                      [16, 32, 3, 2]                \n",
            "  2                -1  1      4800  models.common.C3                        [32, 32, 1]                   \n",
            "  3                -1  1     18560  models.common.Conv                      [32, 64, 3, 2]                \n",
            "  4                -1  2     29184  models.common.C3                        [64, 64, 2]                   \n",
            "  5                -1  1     73984  models.common.Conv                      [64, 128, 3, 2]               \n",
            "  6                -1  3    156928  models.common.C3                        [128, 128, 3]                 \n",
            "  7                -1  1    295424  models.common.Conv                      [128, 256, 3, 2]              \n",
            "  8                -1  1    296448  models.common.C3                        [256, 256, 1]                 \n",
            "  9                -1  1    164608  models.common.SPPF                      [256, 256, 5]                 \n",
            " 10                -1  1     33024  models.common.Conv                      [256, 128, 1, 1]              \n",
            " 11                -1  1         0  torch.nn.modules.upsampling.Upsample    [None, 2, 'nearest']          \n",
            " 12           [-1, 6]  1         0  models.common.Concat                    [1]                           \n",
            " 13                -1  1     90880  models.common.C3                        [256, 128, 1, False]          \n",
            " 14                -1  1      8320  models.common.Conv                      [128, 64, 1, 1]               \n",
            " 15                -1  1         0  torch.nn.modules.upsampling.Upsample    [None, 2, 'nearest']          \n",
            " 16           [-1, 4]  1         0  models.common.Concat                    [1]                           \n",
            " 17                -1  1     22912  models.common.C3                        [128, 64, 1, False]           \n",
            " 18                -1  1     36992  models.common.Conv                      [64, 64, 3, 2]                \n",
            " 19          [-1, 14]  1         0  models.common.Concat                    [1]                           \n",
            " 20                -1  1     74496  models.common.C3                        [128, 128, 1, False]          \n",
            " 21                -1  1    147712  models.common.Conv                      [128, 128, 3, 2]              \n",
            " 22          [-1, 10]  1         0  models.common.Concat                    [1]                           \n",
            " 23                -1  1    296448  models.common.C3                        [256, 256, 1, False]          \n",
            " 24      [17, 20, 23]  1     13530  models.yolo.Detect                      [5, [[10, 13, 16, 30, 33, 23], [30, 61, 62, 45, 59, 119], [116, 90, 156, 198, 373, 326]], [64, 128, 256]]\n",
            "Model Summary: 270 layers, 1770682 parameters, 1770682 gradients, 4.2 GFLOPs\n",
            "\n",
            "Transferred 348/349 items from weights/yolov5n_best.pt\n",
            "load t-model from weights/yolov5l_best.pt\n",
            "Scaled weight_decay = 0.0005\n",
            "\u001b[34m\u001b[1moptimizer:\u001b[0m SGD with parameter groups 57 weight (no decay), 69 weight, 69 bias\n",
            "\u001b[34m\u001b[1malbumentations: \u001b[0mBlur(always_apply=False, p=0.01, blur_limit=(3, 7)), MedianBlur(always_apply=False, p=0.01, blur_limit=(3, 7)), ToGray(always_apply=False, p=0.01), CLAHE(always_apply=False, p=0.01, clip_limit=(1, 4.0), tile_grid_size=(8, 8))\n",
            "\u001b[34m\u001b[1mtrain: \u001b[0mScanning '/content/drive/MyDrive/yolo-distillation/data/dataset/labels/train.cache' images and labels... 991 found, 0 missing, 16 empty, 0 corrupt: 100% 991/991 [00:00<?, ?it/s]\n",
            "\u001b[34m\u001b[1mval: \u001b[0mScanning '/content/drive/MyDrive/yolo-distillation/data/dataset/labels/val.cache' images and labels... 286 found, 0 missing, 7 empty, 0 corrupt: 100% 286/286 [00:00<?, ?it/s]\n",
            "Plotting labels to runs/train/yolov5n-distilled/labels.jpg... \n",
            "\n",
            "\u001b[34m\u001b[1mAutoAnchor: \u001b[0m3.64 anchors/target, 1.000 Best Possible Recall (BPR). Current anchors are a good fit to dataset ✅\n",
            "Image sizes 640 train, 640 val\n",
            "Using 2 dataloader workers\n",
            "Logging results to \u001b[1mruns/train/yolov5n-distilled\u001b[0m\n",
            "Distillation loss type: l2\n",
            "Starting training for 50 epochs...\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      0/49     3.82G   0.02653   0.02091     1.874    0.3464        44       640: 100% 62/62 [00:41<00:00,  1.48it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.77it/s]\n",
            "                 all        286        396      0.719      0.669      0.732      0.451\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      1/49     4.17G   0.02714   0.02261     1.839    0.3154        53       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.41it/s]\n",
            "                 all        286        396      0.769      0.642      0.711      0.441\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      2/49     4.17G   0.02773   0.02284     1.818    0.3198        59       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.51it/s]\n",
            "                 all        286        396      0.713      0.652      0.708      0.422\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      3/49     4.17G     0.029    0.0225     1.748    0.3047        49       640: 100% 62/62 [00:37<00:00,  1.64it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.51it/s]\n",
            "                 all        286        396      0.683      0.669      0.703      0.426\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      4/49     4.17G   0.02924   0.02294     1.759     0.317        47       640: 100% 62/62 [00:37<00:00,  1.63it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.41it/s]\n",
            "                 all        286        396      0.722      0.609      0.684      0.408\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      5/49     4.52G   0.02963   0.02296     1.765    0.3239        46       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.53it/s]\n",
            "                 all        286        396      0.785      0.611      0.711      0.426\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      6/49     4.55G   0.02962   0.02373     1.753    0.3203        49       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.67it/s]\n",
            "                 all        286        396      0.777       0.61      0.695      0.385\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      7/49     4.55G   0.03131   0.02357     1.775    0.3297        61       640: 100% 62/62 [00:38<00:00,  1.63it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:05<00:00,  1.80it/s]\n",
            "                 all        286        396      0.636      0.632      0.662      0.363\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      8/49     4.55G   0.03336   0.02368     1.779    0.3212        58       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.64it/s]\n",
            "                 all        286        396      0.678      0.632       0.67      0.372\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      9/49     4.55G   0.03386   0.02415     1.811    0.3511        46       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.52it/s]\n",
            "                 all        286        396      0.691      0.576      0.637      0.352\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     10/49     4.55G   0.03473   0.02428     1.823    0.3418        54       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.84it/s]\n",
            "                 all        286        396      0.741       0.58      0.643      0.346\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     11/49     4.55G   0.03484   0.02532     1.857    0.3626        54       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.54it/s]\n",
            "                 all        286        396      0.736      0.481      0.587      0.292\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     12/49     4.55G   0.03524   0.02535     1.857    0.3697        37       640: 100% 62/62 [00:37<00:00,  1.64it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.61it/s]\n",
            "                 all        286        396      0.563       0.59      0.611      0.311\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     13/49     4.87G   0.03805    0.0266     1.936    0.3833        46       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.44it/s]\n",
            "                 all        286        396      0.634      0.529      0.559      0.267\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     14/49     4.89G   0.03899   0.02773     2.007    0.4267        39       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.49it/s]\n",
            "                 all        286        396      0.584      0.497      0.473      0.233\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     15/49     4.89G    0.0391   0.02836     2.048    0.4258        51       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.83it/s]\n",
            "                 all        286        396       0.55      0.478      0.459      0.202\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     16/49     4.89G    0.0387    0.0297     2.133    0.4712        48       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.47it/s]\n",
            "                 all        286        396      0.416       0.33      0.274      0.104\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     17/49     4.89G   0.03861   0.03251     2.277     0.521        45       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.49it/s]\n",
            "                 all        286        396       0.49      0.423      0.358      0.143\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     18/49     4.89G   0.03937   0.03149     2.253    0.4996        46       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.68it/s]\n",
            "                 all        286        396      0.433      0.508      0.431      0.208\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     19/49     4.89G   0.03716   0.03023     2.095    0.4624        51       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.89it/s]\n",
            "                 all        286        396      0.585      0.488      0.517      0.256\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     20/49     4.89G   0.03749   0.02886     2.047     0.446        59       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.65it/s]\n",
            "                 all        286        396      0.434       0.47      0.437      0.209\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     21/49     4.89G   0.03768     0.029     2.059    0.4329        58       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.52it/s]\n",
            "                 all        286        396      0.592      0.453      0.477      0.215\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     22/49     4.89G   0.03655   0.02913     2.011    0.4351        45       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396      0.602      0.527      0.569       0.28\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     23/49     4.89G   0.03552   0.02832     1.968    0.4088        33       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.73it/s]\n",
            "                 all        286        396        0.5      0.525      0.508      0.252\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     24/49     4.89G   0.03584    0.0284     1.937    0.4014        47       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.83it/s]\n",
            "                 all        286        396      0.612      0.555      0.572      0.296\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     25/49     4.89G   0.03507   0.02642     1.853    0.3809        45       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.43it/s]\n",
            "                 all        286        396      0.632      0.628       0.62      0.314\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     26/49     4.89G   0.03402   0.02587     1.795    0.3667        55       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.56it/s]\n",
            "                 all        286        396      0.649      0.545      0.597      0.321\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     27/49     4.89G   0.03397   0.02547     1.781    0.3544        47       640: 100% 62/62 [00:43<00:00,  1.44it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.41it/s]\n",
            "                 all        286        396      0.696      0.575       0.63      0.349\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     28/49     4.89G   0.03293   0.02512     1.747     0.359        39       640: 100% 62/62 [00:40<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.78it/s]\n",
            "                 all        286        396      0.547      0.642      0.613      0.314\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     29/49     4.89G   0.03345   0.02516     1.775    0.3548        48       640: 100% 62/62 [00:40<00:00,  1.53it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.27it/s]\n",
            "                 all        286        396      0.623      0.656      0.631      0.328\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     30/49     4.89G   0.03192   0.02448     1.699    0.3351        45       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.50it/s]\n",
            "                 all        286        396      0.611      0.656      0.643      0.363\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     31/49     4.89G    0.0313   0.02319      1.63     0.311        44       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.54it/s]\n",
            "                 all        286        396      0.686      0.567      0.639      0.351\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     32/49     4.89G   0.03253   0.02283     1.608    0.2989        50       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:05<00:00,  1.77it/s]\n",
            "                 all        286        396      0.678      0.621      0.668      0.386\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     33/49     4.89G   0.03147   0.02286     1.616    0.3024        48       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.52it/s]\n",
            "                 all        286        396      0.718       0.62      0.668      0.372\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     34/49     4.89G   0.03048    0.0213     1.532    0.2824        40       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.36it/s]\n",
            "                 all        286        396      0.799      0.558      0.675      0.364\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     35/49     4.89G   0.03071   0.02143     1.523    0.2689        64       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.63it/s]\n",
            "                 all        286        396      0.662      0.668      0.692      0.373\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     36/49     4.89G   0.03005   0.02134     1.518    0.2806        49       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.62it/s]\n",
            "                 all        286        396       0.72      0.651      0.691      0.399\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     37/49     4.89G    0.0284   0.02056     1.481    0.2638        44       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396       0.62      0.699      0.686      0.393\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     38/49     4.89G   0.02854   0.02141     1.448    0.2587        53       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.57it/s]\n",
            "                 all        286        396       0.68      0.704       0.73      0.435\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     39/49     4.89G   0.02887   0.02014      1.44    0.2514        52       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.64it/s]\n",
            "                 all        286        396      0.678      0.648      0.702      0.417\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     40/49     4.89G   0.02794   0.02013     1.411    0.2489        45       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.48it/s]\n",
            "                 all        286        396      0.737      0.639      0.708      0.417\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     41/49     4.89G   0.02731   0.02004     1.389    0.2435        31       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.41it/s]\n",
            "                 all        286        396      0.703      0.637      0.699      0.419\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     42/49     4.89G   0.02767   0.01959     1.384     0.239        63       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.83it/s]\n",
            "                 all        286        396      0.734      0.676      0.723      0.423\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     43/49     4.89G   0.02655   0.01919     1.336    0.2305        53       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.60it/s]\n",
            "                 all        286        396      0.661       0.73      0.729      0.441\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     44/49     4.89G   0.02665   0.01948     1.336    0.2304        44       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.85it/s]\n",
            "                 all        286        396      0.775      0.657      0.732      0.449\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     45/49      4.9G   0.02669    0.0191     1.325    0.2274        43       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.38it/s]\n",
            "                 all        286        396      0.795      0.663       0.75      0.453\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     46/49      4.9G   0.02662   0.01821     1.302    0.2125        45       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.20it/s]\n",
            "                 all        286        396       0.72      0.727      0.762      0.464\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     47/49      4.9G   0.02623   0.01899     1.303    0.2195        50       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.40it/s]\n",
            "                 all        286        396       0.75      0.728      0.754      0.445\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     48/49      4.9G   0.02552   0.01852      1.29    0.2152        42       640: 100% 62/62 [00:38<00:00,  1.63it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.50it/s]\n",
            "                 all        286        396      0.746      0.693      0.746      0.452\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     49/49      4.9G   0.02584   0.01805      1.29    0.2075        50       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.36it/s]\n",
            "                 all        286        396      0.739      0.704      0.754      0.457\n",
            "\n",
            "50 epochs completed in 0.628 hours.\n",
            "Optimizer stripped from runs/train/yolov5n-distilled/weights/last.pt, 275.7MB\n",
            "Optimizer stripped from runs/train/yolov5n-distilled/weights/best.pt, 275.7MB\n",
            "\n",
            "Validating runs/train/yolov5n-distilled/weights/best.pt...\n",
            "Fusing layers... \n",
            "Model Summary: 232 layers, 26887978 parameters, 0 gradients, 4.1 GFLOPs\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.02it/s]\n",
            "                 all        286        396      0.749        0.7      0.762      0.464\n",
            "                 cat        286         82      0.812      0.793      0.785      0.466\n",
            "                 cow        286         44      0.705      0.705      0.788      0.532\n",
            "               sheep        286         60      0.663      0.533      0.657      0.374\n",
            "                 dog        286        122       0.72      0.652      0.711      0.408\n",
            "               horse        286         88      0.847      0.816       0.87      0.541\n",
            "Results saved to \u001b[1mruns/train/yolov5n-distilled\u001b[0m\n"
          ]
        }
      ],
      "source": [
        "!python train_dist.py --weights weights/yolov5n_best.pt --cfg models/yolov5n_animal.yaml --data data/VOC_animal.yaml --batch-size 16 --epochs 50 --workers 2 --t_weights weights/yolov5l_best.pt --hyp data/hyps/hyp.scratch-low-distillation.yaml --name yolov5n-distilled --distill_cfg configs/distill_config2.py"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "hKGiZBs8qHU2"
      },
      "outputs": [],
      "source": [
        "!python val.py --data data/VOC_animal.yaml --weights runs/train/yolov5n-distilled/weights/best.pt --batch-size 16"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 5,
      "metadata": {
        "id": "6nAsDmMw9Etr"
      },
      "outputs": [],
      "source": [
        "!rm -rf /content/drive/MyDrive/yolo-distillation/runs/train/yolov5n-distilled3"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "!rm -rf /content/drive/MyDrive/yolo-distillation/runs/val/exp19"
      ],
      "metadata": {
        "id": "QU0KeWEUl5xO"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "ykhJqfEOgYYP"
      },
      "outputs": [],
      "source": [
        "!python train_dist.py --weights weights/yolov5n_best.pt --cfg models/yolov5n_animal.yaml --data data/VOC_animal.yaml --batch-size 16 --epochs 24 --workers 2 --t_weights weights/yolov5l_best.pt --hyp data/hyps/hyp.scratch-low-distillation.yaml --name yolov5n-distilled --distill_cfg configs/distill_config2.py"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 7,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "v79J18P4r6S8",
        "outputId": "f08f60a6-da72-4e1c-be00-c42656920dbc"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\u001b[34m\u001b[1mval: \u001b[0mdata=data/VOC_animal.yaml, weights=['runs/train/yolov5n-distilled3/weights/best.pt'], batch_size=16, imgsz=640, conf_thres=0.001, iou_thres=0.6, task=val, device=, workers=8, single_cls=False, augment=False, verbose=False, save_txt=False, save_hybrid=False, save_conf=False, save_json=False, project=runs/val, name=exp, exist_ok=False, half=False, dnn=False\n",
            "YOLOv5 🚀 3f3185b torch 1.12.1+cu113 CUDA:0 (Tesla T4, 15110MiB)\n",
            "\n",
            "Generating grammar tables from /usr/lib/python3.7/lib2to3/Grammar.txt\n",
            "Generating grammar tables from /usr/lib/python3.7/lib2to3/PatternGrammar.txt\n",
            "Fusing layers... \n",
            "Model Summary: 232 layers, 26887978 parameters, 0 gradients, 4.1 GFLOPs\n",
            "\u001b[34m\u001b[1mval: \u001b[0mScanning '/content/drive/.shortcut-targets-by-id/1-CXP81xST-t3P2hLZQLjfipgHlqh7k86/yolo-distillation/data/dataset/labels/val.cache' images and labels... 286 found, 0 missing, 7 empty, 0 corrupt: 100% 286/286 [00:00<?, ?it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 18/18 [00:04<00:00,  3.75it/s]\n",
            "                 all        286        396      0.795        0.8      0.821      0.535\n",
            "                 cat        286         82      0.795      0.793      0.797      0.488\n",
            "                 cow        286         44      0.791      0.862      0.849      0.625\n",
            "               sheep        286         60      0.794      0.667      0.751      0.455\n",
            "                 dog        286        122        0.7      0.787      0.782      0.484\n",
            "               horse        286         88      0.897      0.891      0.924      0.622\n",
            "Speed: 0.2ms pre-process, 3.4ms inference, 2.7ms NMS per image at shape (16, 3, 640, 640)\n",
            "Results saved to \u001b[1mruns/val/exp2\u001b[0m\n"
          ]
        }
      ],
      "source": [
        "!python val.py --data data/VOC_animal.yaml --weights runs/train/yolov5n-distilled3/weights/best.pt --batch-size 16"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 6,
      "metadata": {
        "id": "-MosGWvuQQcN",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "917419de-0060-40e8-ba4c-26279e33612a"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Generating grammar tables from /usr/lib/python3.7/lib2to3/Grammar.txt\n",
            "Generating grammar tables from /usr/lib/python3.7/lib2to3/PatternGrammar.txt\n",
            "\u001b[34m\u001b[1mtrain_dist: \u001b[0mweights=weights/yolov5n_best.pt, t_weights=weights/yolov5l_best.pt, dist_loss=l2, temperature=20, distill_cfg=configs/distill_config2.py, cfg=models/yolov5n_animal.yaml, data=data/VOC_animal.yaml, hyp=data/hyps/hyp.scratch-low-distillation.yaml, epochs=100, batch_size=16, imgsz=640, rect=False, resume=False, nosave=False, noval=False, noautoanchor=False, evolve=None, bucket=, cache=None, image_weights=False, device=, multi_scale=False, single_cls=False, optimizer=SGD, sync_bn=False, workers=2, project=runs/train, name=yolov5n-distilled, exist_ok=False, quad=False, cos_lr=False, label_smoothing=0.0, patience=100, freeze=[0], save_period=-1, local_rank=-1, entity=None, upload_dataset=False, bbox_interval=-1, artifact_alias=latest\n",
            "\u001b[34m\u001b[1mgithub: \u001b[0mup to date with https://gitee.com/zfhwang/yolo-distillation ✅\n",
            "YOLOv5 🚀 3f3185b torch 1.12.1+cu113 CUDA:0 (Tesla T4, 15110MiB)\n",
            "\n",
            "\u001b[34m\u001b[1mhyperparameters: \u001b[0mlr0=0.01, lrf=0.01, momentum=0.9, weight_decay=0.0001, warmup_epochs=3.0, warmup_momentum=0.8, warmup_bias_lr=0.1, box=0.05, cls=0.5, cls_pw=1.0, obj=1.0, obj_pw=1.0, iou_t=0.2, anchor_t=4.0, fl_gamma=0.0, hsv_h=0.015, hsv_s=0.7, hsv_v=0.4, degrees=0.0, translate=0.1, scale=0.5, shear=0.0, perspective=0.0, flipud=0.0, fliplr=0.5, mosaic=1.0, mixup=0.0, copy_paste=0.0, dist=0.5\n",
            "\u001b[34m\u001b[1mWeights & Biases: \u001b[0mrun 'pip install wandb' to automatically track and visualize YOLOv5 🚀 runs (RECOMMENDED)\n",
            "\u001b[34m\u001b[1mTensorBoard: \u001b[0mStart with 'tensorboard --logdir runs/train', view at http://localhost:6006/\n",
            "\n",
            "                 from  n    params  module                                  arguments                     \n",
            "  0                -1  1      1760  models.common.Conv                      [3, 16, 6, 2, 2]              \n",
            "  1                -1  1      4672  models.common.Conv                      [16, 32, 3, 2]                \n",
            "  2                -1  1      4800  models.common.C3                        [32, 32, 1]                   \n",
            "  3                -1  1     18560  models.common.Conv                      [32, 64, 3, 2]                \n",
            "  4                -1  2     29184  models.common.C3                        [64, 64, 2]                   \n",
            "  5                -1  1     73984  models.common.Conv                      [64, 128, 3, 2]               \n",
            "  6                -1  3    156928  models.common.C3                        [128, 128, 3]                 \n",
            "  7                -1  1    295424  models.common.Conv                      [128, 256, 3, 2]              \n",
            "  8                -1  1    296448  models.common.C3                        [256, 256, 1]                 \n",
            "  9                -1  1    164608  models.common.SPPF                      [256, 256, 5]                 \n",
            " 10                -1  1     33024  models.common.Conv                      [256, 128, 1, 1]              \n",
            " 11                -1  1         0  torch.nn.modules.upsampling.Upsample    [None, 2, 'nearest']          \n",
            " 12           [-1, 6]  1         0  models.common.Concat                    [1]                           \n",
            " 13                -1  1     90880  models.common.C3                        [256, 128, 1, False]          \n",
            " 14                -1  1      8320  models.common.Conv                      [128, 64, 1, 1]               \n",
            " 15                -1  1         0  torch.nn.modules.upsampling.Upsample    [None, 2, 'nearest']          \n",
            " 16           [-1, 4]  1         0  models.common.Concat                    [1]                           \n",
            " 17                -1  1     22912  models.common.C3                        [128, 64, 1, False]           \n",
            " 18                -1  1     36992  models.common.Conv                      [64, 64, 3, 2]                \n",
            " 19          [-1, 14]  1         0  models.common.Concat                    [1]                           \n",
            " 20                -1  1     74496  models.common.C3                        [128, 128, 1, False]          \n",
            " 21                -1  1    147712  models.common.Conv                      [128, 128, 3, 2]              \n",
            " 22          [-1, 10]  1         0  models.common.Concat                    [1]                           \n",
            " 23                -1  1    296448  models.common.C3                        [256, 256, 1, False]          \n",
            " 24      [17, 20, 23]  1     13530  models.yolo.Detect                      [5, [[10, 13, 16, 30, 33, 23], [30, 61, 62, 45, 59, 119], [116, 90, 156, 198, 373, 326]], [64, 128, 256]]\n",
            "Model Summary: 270 layers, 1770682 parameters, 1770682 gradients, 4.2 GFLOPs\n",
            "\n",
            "Transferred 348/349 items from weights/yolov5n_best.pt\n",
            "load t-model from weights/yolov5l_best.pt\n",
            "Scaled weight_decay = 0.0001\n",
            "\u001b[34m\u001b[1moptimizer:\u001b[0m SGD with parameter groups 57 weight (no decay), 69 weight, 69 bias\n",
            "\u001b[34m\u001b[1malbumentations: \u001b[0mBlur(always_apply=False, p=0.01, blur_limit=(3, 7)), MedianBlur(always_apply=False, p=0.01, blur_limit=(3, 7)), ToGray(always_apply=False, p=0.01), CLAHE(always_apply=False, p=0.01, clip_limit=(1, 4.0), tile_grid_size=(8, 8))\n",
            "\u001b[34m\u001b[1mtrain: \u001b[0mScanning '/content/drive/.shortcut-targets-by-id/1-CXP81xST-t3P2hLZQLjfipgHlqh7k86/yolo-distillation/data/dataset/labels/train.cache' images and labels... 991 found, 0 missing, 16 empty, 0 corrupt: 100% 991/991 [00:00<?, ?it/s]\n",
            "\u001b[34m\u001b[1mval: \u001b[0mScanning '/content/drive/.shortcut-targets-by-id/1-CXP81xST-t3P2hLZQLjfipgHlqh7k86/yolo-distillation/data/dataset/labels/val.cache' images and labels... 286 found, 0 missing, 7 empty, 0 corrupt: 100% 286/286 [00:00<?, ?it/s]\n",
            "Plotting labels to runs/train/yolov5n-distilled3/labels.jpg... \n",
            "\n",
            "\u001b[34m\u001b[1mAutoAnchor: \u001b[0m3.64 anchors/target, 1.000 Best Possible Recall (BPR). Current anchors are a good fit to dataset ✅\n",
            "Image sizes 640 train, 640 val\n",
            "Using 2 dataloader workers\n",
            "Logging results to \u001b[1mruns/train/yolov5n-distilled3\u001b[0m\n",
            "Distillation loss type: l2\n",
            "Starting training for 100 epochs...\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      0/99     3.86G   0.02644   0.01808  0.008403        44       640: 100% 62/62 [00:44<00:00,  1.41it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.28it/s]\n",
            "                 all        286        396      0.694      0.717      0.736      0.458\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      1/99      4.2G   0.02723   0.01881  0.008752        53       640: 100% 62/62 [00:42<00:00,  1.47it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.36it/s]\n",
            "                 all        286        396      0.671      0.691      0.713      0.433\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      2/99      4.2G   0.02824   0.01891  0.009459        59       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396      0.735      0.692      0.742       0.45\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      3/99      4.2G   0.02957   0.01881  0.008693        49       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.47it/s]\n",
            "                 all        286        396      0.717      0.666      0.726      0.443\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      4/99      4.2G   0.03001   0.01883  0.009353        47       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.30it/s]\n",
            "                 all        286        396      0.745      0.694      0.737      0.445\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      5/99     4.55G   0.03073    0.0188   0.01005        46       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.32it/s]\n",
            "                 all        286        396      0.774      0.654      0.726      0.433\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      6/99     4.58G   0.03088   0.01928   0.01016        49       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.52it/s]\n",
            "                 all        286        396      0.755       0.69      0.736      0.443\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      7/99     4.58G   0.03215   0.01907   0.01013        61       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.46it/s]\n",
            "                 all        286        396      0.685      0.709      0.753      0.431\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      8/99     4.58G    0.0345   0.01904   0.01005        58       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.41it/s]\n",
            "                 all        286        396      0.662      0.639      0.634      0.341\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      9/99     4.58G   0.03527   0.01924    0.0104        46       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.37it/s]\n",
            "                 all        286        396      0.603      0.704      0.668      0.379\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     10/99     4.61G   0.03645    0.0193   0.01092        54       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.57it/s]\n",
            "                 all        286        396      0.597      0.308      0.274       0.12\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     11/99     4.61G   0.03771   0.02029   0.01135        54       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.34it/s]\n",
            "                 all        286        396      0.656      0.588      0.648      0.347\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     12/99     4.61G   0.03597   0.01926   0.01093        37       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396      0.697      0.646      0.684      0.364\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     13/99     4.93G   0.03879   0.02008   0.01094        46       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.53it/s]\n",
            "                 all        286        396      0.615      0.611      0.634      0.312\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     14/99     4.93G    0.0393   0.02091   0.01217        39       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.30it/s]\n",
            "                 all        286        396      0.569      0.449      0.429      0.209\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     15/99     4.93G    0.0418   0.02079   0.01381        51       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.48it/s]\n",
            "                 all        286        396      0.552      0.437      0.445      0.206\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     16/99     4.93G   0.04013   0.02053   0.01475        48       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.65it/s]\n",
            "                 all        286        396      0.554      0.495      0.487      0.223\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     17/99     4.93G   0.03902   0.02128   0.01448        45       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.57it/s]\n",
            "                 all        286        396      0.627      0.584       0.61      0.329\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     18/99     4.93G   0.03953    0.0208   0.01497        46       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.31it/s]\n",
            "                 all        286        396      0.496       0.45      0.463      0.233\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     19/99     4.93G   0.03848   0.02057   0.01234        51       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.49it/s]\n",
            "                 all        286        396      0.756      0.506      0.596      0.315\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     20/99     4.93G   0.03825   0.01973   0.01128        59       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.36it/s]\n",
            "                 all        286        396       0.73      0.632      0.694      0.378\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     21/99     4.93G   0.03771   0.01992   0.01142        58       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.21it/s]\n",
            "                 all        286        396      0.672      0.599      0.656      0.361\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     22/99     4.93G   0.03707   0.02047    0.0101        45       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.48it/s]\n",
            "                 all        286        396       0.66      0.664      0.679      0.374\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     23/99     4.93G   0.03593   0.01978   0.01101        33       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.26it/s]\n",
            "                 all        286        396      0.663      0.654      0.654      0.363\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     24/99     4.93G   0.03673   0.02024   0.01102        47       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.41it/s]\n",
            "                 all        286        396      0.718      0.598      0.658      0.358\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     25/99     4.93G   0.03629   0.01926  0.009728        45       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.44it/s]\n",
            "                 all        286        396      0.646      0.627      0.654      0.369\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     26/99     4.93G   0.03517   0.01936  0.009383        55       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.34it/s]\n",
            "                 all        286        396      0.658      0.646      0.653      0.372\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     27/99     4.93G   0.03474   0.01941  0.009304        47       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.25it/s]\n",
            "                 all        286        396      0.677      0.656      0.685      0.394\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     28/99     4.93G   0.03446   0.01921  0.008651        39       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.88it/s]\n",
            "                 all        286        396      0.744      0.629      0.712       0.41\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     29/99     4.93G   0.03538   0.01908  0.009281        48       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.21it/s]\n",
            "                 all        286        396      0.661      0.651      0.677       0.38\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     30/99     4.93G   0.03346   0.01913  0.009547        45       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.26it/s]\n",
            "                 all        286        396      0.661      0.655      0.702      0.405\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     31/99     4.93G   0.03238   0.01887  0.009158        44       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.30it/s]\n",
            "                 all        286        396      0.689      0.661      0.696      0.399\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     32/99     4.93G   0.03415   0.01904  0.008501        50       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.60it/s]\n",
            "                 all        286        396      0.696      0.704      0.731      0.425\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     33/99     4.93G   0.03314   0.01912  0.009235        48       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.54it/s]\n",
            "                 all        286        396      0.715      0.658      0.707      0.408\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     34/99     4.93G   0.03293   0.01805  0.008867        40       640: 100% 62/62 [00:40<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.63it/s]\n",
            "                 all        286        396      0.706      0.672      0.714      0.391\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     35/99     4.93G   0.03291   0.01837  0.009406        64       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.58it/s]\n",
            "                 all        286        396      0.701      0.703      0.752      0.426\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     36/99     4.93G   0.03266   0.01864  0.009498        49       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.55it/s]\n",
            "                 all        286        396      0.757      0.663      0.718      0.428\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     37/99     4.93G   0.03178   0.01828   0.01023        44       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.44it/s]\n",
            "                 all        286        396      0.725      0.657      0.727      0.416\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     38/99     4.93G   0.03239   0.01896  0.007639        53       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.37it/s]\n",
            "                 all        286        396      0.709      0.674      0.708      0.411\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     39/99     4.93G   0.03223   0.01808  0.008735        52       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.23it/s]\n",
            "                 all        286        396      0.745       0.72      0.755      0.435\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     40/99     4.93G   0.03163    0.0183  0.008838        45       640: 100% 62/62 [00:40<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.66it/s]\n",
            "                 all        286        396       0.71       0.72      0.748      0.447\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     41/99     4.93G   0.03122   0.01818   0.00732        31       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.63it/s]\n",
            "                 all        286        396      0.701      0.719       0.74      0.446\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     42/99     4.93G   0.03179   0.01823  0.008328        63       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.37it/s]\n",
            "                 all        286        396      0.756      0.677       0.74      0.444\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     43/99     4.93G   0.03085   0.01788  0.007446        53       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.52it/s]\n",
            "                 all        286        396      0.726      0.636      0.705      0.427\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     44/99     4.93G   0.03106   0.01824  0.007232        44       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.52it/s]\n",
            "                 all        286        396      0.752      0.696      0.736      0.444\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     45/99     4.93G   0.03107   0.01818  0.007989        43       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.48it/s]\n",
            "                 all        286        396      0.714      0.708      0.717      0.435\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     46/99     4.93G   0.03127   0.01757  0.008692        45       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.59it/s]\n",
            "                 all        286        396      0.754      0.683      0.739      0.441\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     47/99     4.93G   0.03104   0.01831  0.006644        50       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.62it/s]\n",
            "                 all        286        396      0.778      0.687       0.76       0.46\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     48/99     4.93G   0.03033   0.01782  0.006954        42       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396      0.699       0.73      0.734      0.436\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     49/99     4.93G   0.03106   0.01741    0.0079        50       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.36it/s]\n",
            "                 all        286        396      0.724      0.737      0.764      0.463\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     50/99     4.93G   0.03042   0.01795  0.007581        52       640: 100% 62/62 [00:40<00:00,  1.52it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.74it/s]\n",
            "                 all        286        396      0.722      0.685      0.717       0.44\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     51/99     4.93G   0.03037   0.01738  0.007545        40       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.52it/s]\n",
            "                 all        286        396      0.755      0.732      0.769      0.471\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     52/99     4.93G   0.03002   0.01761  0.008343        39       640: 100% 62/62 [00:40<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.33it/s]\n",
            "                 all        286        396      0.729      0.692      0.729      0.455\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     53/99     4.93G   0.02994   0.01744  0.008058        36       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.51it/s]\n",
            "                 all        286        396       0.74      0.714      0.769      0.482\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     54/99     4.93G   0.02852   0.01835  0.006904        41       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.64it/s]\n",
            "                 all        286        396      0.803      0.701      0.768      0.468\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     55/99     4.93G   0.02907   0.01695  0.006088        51       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.53it/s]\n",
            "                 all        286        396      0.788      0.715      0.783      0.489\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     56/99     4.93G   0.02861   0.01719  0.006743        35       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.35it/s]\n",
            "                 all        286        396      0.746      0.731      0.763      0.474\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     57/99     4.93G   0.02847    0.0175  0.007622        55       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.32it/s]\n",
            "                 all        286        396      0.713      0.776       0.77      0.482\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     58/99     4.93G   0.02938   0.01776  0.006669        30       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.35it/s]\n",
            "                 all        286        396       0.74      0.753      0.773      0.482\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     59/99     4.93G   0.02931   0.01737  0.006675        45       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.48it/s]\n",
            "                 all        286        396      0.785      0.717      0.772      0.481\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     60/99     4.93G   0.02942   0.01773  0.006094        57       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.37it/s]\n",
            "                 all        286        396      0.804      0.727      0.799      0.489\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     61/99     4.93G   0.02821   0.01698   0.00555        39       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.55it/s]\n",
            "                 all        286        396      0.791      0.709      0.775      0.493\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     62/99     4.93G   0.02887   0.01748  0.006839        45       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.39it/s]\n",
            "                 all        286        396      0.786      0.727      0.792      0.502\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     63/99     4.93G   0.02856   0.01767   0.00666        40       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.35it/s]\n",
            "                 all        286        396      0.763      0.704      0.761      0.483\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     64/99     4.93G    0.0275   0.01771  0.006406        54       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.35it/s]\n",
            "                 all        286        396      0.824      0.697      0.791      0.505\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     65/99     4.93G   0.02797   0.01695   0.00622        38       640: 100% 62/62 [00:40<00:00,  1.53it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.31it/s]\n",
            "                 all        286        396      0.755      0.737      0.777      0.494\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     66/99     4.93G   0.02811   0.01739  0.006529        44       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.48it/s]\n",
            "                 all        286        396      0.822      0.757      0.816      0.526\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     67/99     4.93G    0.0271   0.01723  0.006211        50       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.32it/s]\n",
            "                 all        286        396      0.775      0.759      0.791      0.502\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     68/99     4.93G   0.02737   0.01649   0.00597        42       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.45it/s]\n",
            "                 all        286        396      0.753      0.748      0.792      0.495\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     69/99     4.93G   0.02694   0.01721  0.006234        48       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.41it/s]\n",
            "                 all        286        396      0.834      0.718      0.795       0.52\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     70/99     4.93G   0.02721   0.01669  0.006614        40       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.58it/s]\n",
            "                 all        286        396      0.784      0.733      0.787      0.499\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     71/99     4.93G   0.02797   0.01681  0.006788        41       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.31it/s]\n",
            "                 all        286        396      0.785      0.695      0.758      0.479\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     72/99     4.93G   0.02647   0.01682  0.006034        56       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.34it/s]\n",
            "                 all        286        396      0.767      0.735      0.796      0.513\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     73/99     4.93G   0.02691   0.01686  0.005971        57       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.61it/s]\n",
            "                 all        286        396      0.829      0.703      0.795      0.498\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     74/99     4.93G   0.02619   0.01667  0.005914        44       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.39it/s]\n",
            "                 all        286        396      0.832       0.69      0.803      0.515\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     75/99     4.93G   0.02617   0.01637  0.005912        47       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.52it/s]\n",
            "                 all        286        396       0.78      0.735       0.78      0.503\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     76/99     4.93G   0.02618   0.01735  0.006281        43       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.26it/s]\n",
            "                 all        286        396      0.823      0.714      0.794      0.514\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     77/99     4.93G   0.02596   0.01713  0.005652        46       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.29it/s]\n",
            "                 all        286        396      0.803      0.735      0.794       0.51\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     78/99     4.93G    0.0263   0.01649   0.00599        35       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.56it/s]\n",
            "                 all        286        396      0.833      0.743      0.804      0.514\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     79/99     4.93G   0.02601   0.01657  0.005922        44       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.36it/s]\n",
            "                 all        286        396      0.747      0.763      0.787      0.503\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     80/99     4.93G   0.02611    0.0165  0.006407        44       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.32it/s]\n",
            "                 all        286        396      0.797       0.72      0.785      0.512\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     81/99     4.93G   0.02559   0.01628  0.004907        55       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.71it/s]\n",
            "                 all        286        396      0.835      0.722      0.804      0.518\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     82/99     4.93G   0.02583   0.01619  0.005867        36       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.39it/s]\n",
            "                 all        286        396      0.794      0.762      0.811      0.517\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     83/99     4.93G   0.02578    0.0165  0.005951        41       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.27it/s]\n",
            "                 all        286        396      0.816      0.727      0.795      0.519\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     84/99     4.93G   0.02473   0.01646  0.004765        49       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.45it/s]\n",
            "                 all        286        396      0.807      0.724      0.794      0.519\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     85/99     4.93G   0.02562   0.01682  0.005444        39       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.39it/s]\n",
            "                 all        286        396      0.817      0.771      0.817      0.526\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     86/99     4.93G   0.02542   0.01624  0.005826        51       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.31it/s]\n",
            "                 all        286        396      0.816       0.76      0.808      0.526\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     87/99     4.93G   0.02522   0.01607  0.005993        47       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.74it/s]\n",
            "                 all        286        396       0.76      0.776      0.801      0.518\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     88/99     4.93G   0.02525   0.01544  0.004579        55       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.46it/s]\n",
            "                 all        286        396      0.852      0.735       0.82       0.53\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     89/99     4.93G   0.02528   0.01604  0.004857        45       640: 100% 62/62 [00:40<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.45it/s]\n",
            "                 all        286        396      0.836      0.741      0.802      0.512\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     90/99     4.93G   0.02446   0.01571   0.00498        40       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.63it/s]\n",
            "                 all        286        396      0.788      0.775      0.809      0.529\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     91/99     4.93G   0.02458   0.01576  0.004823        43       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.39it/s]\n",
            "                 all        286        396      0.813      0.753      0.806      0.522\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     92/99     4.93G   0.02483   0.01612  0.004641        37       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.64it/s]\n",
            "                 all        286        396      0.825      0.719      0.795      0.523\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     93/99     4.93G   0.02438   0.01605  0.004859        56       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396       0.83      0.766      0.806       0.53\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     94/99     4.93G   0.02404   0.01584  0.004888        41       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.38it/s]\n",
            "                 all        286        396      0.823      0.743      0.801      0.523\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     95/99     4.93G   0.02424   0.01567  0.005515        50       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.36it/s]\n",
            "                 all        286        396      0.799      0.782      0.808      0.528\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     96/99     4.93G   0.02432   0.01648  0.004854        38       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.54it/s]\n",
            "                 all        286        396      0.807      0.771      0.804       0.53\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     97/99     4.93G   0.02357   0.01574  0.004417        51       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.22it/s]\n",
            "                 all        286        396      0.772      0.785      0.805       0.53\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     98/99     4.93G   0.02432   0.01583  0.005352        48       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.15it/s]\n",
            "                 all        286        396      0.797      0.796      0.818      0.533\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     99/99     4.93G   0.02465   0.01636  0.004883        37       640: 100% 62/62 [00:41<00:00,  1.50it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.29it/s]\n",
            "                 all        286        396      0.793      0.781      0.812      0.533\n",
            "\n",
            "100 epochs completed in 1.272 hours.\n",
            "Optimizer stripped from runs/train/yolov5n-distilled3/weights/last.pt, 275.7MB\n",
            "Optimizer stripped from runs/train/yolov5n-distilled3/weights/best.pt, 275.7MB\n",
            "\n",
            "Validating runs/train/yolov5n-distilled3/weights/best.pt...\n",
            "Fusing layers... \n",
            "Model Summary: 232 layers, 26887978 parameters, 0 gradients, 4.1 GFLOPs\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.10it/s]\n",
            "                 all        286        396      0.797      0.796      0.818      0.534\n",
            "                 cat        286         82      0.805      0.793      0.807      0.486\n",
            "                 cow        286         44      0.791      0.859      0.849      0.628\n",
            "               sheep        286         60      0.777       0.65      0.736      0.457\n",
            "                 dog        286        122      0.706      0.787      0.774      0.479\n",
            "               horse        286         88      0.908      0.892      0.924      0.619\n",
            "Results saved to \u001b[1mruns/train/yolov5n-distilled3\u001b[0m\n"
          ]
        }
      ],
      "source": [
        "!python train_dist.py --weights weights/yolov5n_best.pt --cfg models/yolov5n_animal.yaml --data data/VOC_animal.yaml --batch-size 16 --epochs 100 --workers 2 --t_weights weights/yolov5l_best.pt --hyp data/hyps/hyp.scratch-low-distillation.yaml --name yolov5n-distilled --distill_cfg configs/distill_config2.py"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 7,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "WmP8xU08oBOr",
        "outputId": "150d634a-27c4-457d-8702-b58062d12e66"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Downloading https://ultralytics.com/assets/Arial.ttf to /root/.config/Ultralytics/Arial.ttf...\n",
            "Generating grammar tables from /usr/lib/python3.7/lib2to3/Grammar.txt\n",
            "Generating grammar tables from /usr/lib/python3.7/lib2to3/PatternGrammar.txt\n",
            "\u001b[34m\u001b[1mtrain_dist: \u001b[0mweights=weights/yolov5n_best.pt, t_weights=weights/yolov5l_best.pt, dist_loss=l2, temperature=20, distill_cfg=configs/distill_config2.py, cfg=models/yolov5n_animal.yaml, data=data/VOC_animal.yaml, hyp=data/hyps/hyp.scratch-low-distillation.yaml, epochs=100, batch_size=16, imgsz=640, rect=False, resume=False, nosave=False, noval=False, noautoanchor=False, evolve=None, bucket=, cache=None, image_weights=False, device=, multi_scale=False, single_cls=False, optimizer=SGD, sync_bn=False, workers=2, project=runs/train, name=yolov5n-distilled, exist_ok=False, quad=False, cos_lr=False, label_smoothing=0.0, patience=100, freeze=[0], save_period=-1, local_rank=-1, entity=None, upload_dataset=False, bbox_interval=-1, artifact_alias=latest\n",
            "Command 'git fetch && git config --get remote.origin.url' timed out after 5 seconds\n",
            "YOLOv5 🚀 3f3185b torch 1.12.1+cu113 CUDA:0 (Tesla T4, 15110MiB)\n",
            "\n",
            "\u001b[34m\u001b[1mhyperparameters: \u001b[0mlr0=0.01, lrf=0.01, momentum=0.9, weight_decay=0.0001, warmup_epochs=3.0, warmup_momentum=0.8, warmup_bias_lr=0.1, box=0.05, cls=0.5, cls_pw=1.0, obj=1.0, obj_pw=1.0, iou_t=0.2, anchor_t=4.0, fl_gamma=0.0, hsv_h=0.015, hsv_s=0.7, hsv_v=0.4, degrees=0.0, translate=0.1, scale=0.5, shear=0.0, perspective=0.0, flipud=0.0, fliplr=0.5, mosaic=1.0, mixup=0.0, copy_paste=0.0, dist=0.5\n",
            "\u001b[34m\u001b[1mWeights & Biases: \u001b[0mrun 'pip install wandb' to automatically track and visualize YOLOv5 🚀 runs (RECOMMENDED)\n",
            "\u001b[34m\u001b[1mTensorBoard: \u001b[0mStart with 'tensorboard --logdir runs/train', view at http://localhost:6006/\n",
            "\n",
            "                 from  n    params  module                                  arguments                     \n",
            "  0                -1  1      1760  models.common.Conv                      [3, 16, 6, 2, 2]              \n",
            "  1                -1  1      4672  models.common.Conv                      [16, 32, 3, 2]                \n",
            "  2                -1  1      4800  models.common.C3                        [32, 32, 1]                   \n",
            "  3                -1  1     18560  models.common.Conv                      [32, 64, 3, 2]                \n",
            "  4                -1  2     29184  models.common.C3                        [64, 64, 2]                   \n",
            "  5                -1  1     73984  models.common.Conv                      [64, 128, 3, 2]               \n",
            "  6                -1  3    156928  models.common.C3                        [128, 128, 3]                 \n",
            "  7                -1  1    295424  models.common.Conv                      [128, 256, 3, 2]              \n",
            "  8                -1  1    296448  models.common.C3                        [256, 256, 1]                 \n",
            "  9                -1  1    164608  models.common.SPPF                      [256, 256, 5]                 \n",
            " 10                -1  1     33024  models.common.Conv                      [256, 128, 1, 1]              \n",
            " 11                -1  1         0  torch.nn.modules.upsampling.Upsample    [None, 2, 'nearest']          \n",
            " 12           [-1, 6]  1         0  models.common.Concat                    [1]                           \n",
            " 13                -1  1     90880  models.common.C3                        [256, 128, 1, False]          \n",
            " 14                -1  1      8320  models.common.Conv                      [128, 64, 1, 1]               \n",
            " 15                -1  1         0  torch.nn.modules.upsampling.Upsample    [None, 2, 'nearest']          \n",
            " 16           [-1, 4]  1         0  models.common.Concat                    [1]                           \n",
            " 17                -1  1     22912  models.common.C3                        [128, 64, 1, False]           \n",
            " 18                -1  1     36992  models.common.Conv                      [64, 64, 3, 2]                \n",
            " 19          [-1, 14]  1         0  models.common.Concat                    [1]                           \n",
            " 20                -1  1     74496  models.common.C3                        [128, 128, 1, False]          \n",
            " 21                -1  1    147712  models.common.Conv                      [128, 128, 3, 2]              \n",
            " 22          [-1, 10]  1         0  models.common.Concat                    [1]                           \n",
            " 23                -1  1    296448  models.common.C3                        [256, 256, 1, False]          \n",
            " 24      [17, 20, 23]  1     13530  models.yolo.Detect                      [5, [[10, 13, 16, 30, 33, 23], [30, 61, 62, 45, 59, 119], [116, 90, 156, 198, 373, 326]], [64, 128, 256]]\n",
            "Model Summary: 270 layers, 1770682 parameters, 1770682 gradients, 4.2 GFLOPs\n",
            "\n",
            "Transferred 348/349 items from weights/yolov5n_best.pt\n",
            "load t-model from weights/yolov5l_best.pt\n",
            "Scaled weight_decay = 0.0001\n",
            "\u001b[34m\u001b[1moptimizer:\u001b[0m SGD with parameter groups 57 weight (no decay), 69 weight, 69 bias\n",
            "\u001b[34m\u001b[1malbumentations: \u001b[0mBlur(always_apply=False, p=0.01, blur_limit=(3, 7)), MedianBlur(always_apply=False, p=0.01, blur_limit=(3, 7)), ToGray(always_apply=False, p=0.01), CLAHE(always_apply=False, p=0.01, clip_limit=(1, 4.0), tile_grid_size=(8, 8))\n",
            "\u001b[34m\u001b[1mtrain: \u001b[0mScanning '/content/drive/.shortcut-targets-by-id/1-CXP81xST-t3P2hLZQLjfipgHlqh7k86/yolo-distillation/data/dataset/labels/train' images and labels...991 found, 0 missing, 16 empty, 0 corrupt: 100% 991/991 [04:24<00:00,  3.75it/s]\n",
            "\u001b[34m\u001b[1mtrain: \u001b[0mNew cache created: /content/drive/.shortcut-targets-by-id/1-CXP81xST-t3P2hLZQLjfipgHlqh7k86/yolo-distillation/data/dataset/labels/train.cache\n",
            "\u001b[34m\u001b[1mval: \u001b[0mScanning '/content/drive/.shortcut-targets-by-id/1-CXP81xST-t3P2hLZQLjfipgHlqh7k86/yolo-distillation/data/dataset/labels/val' images and labels...286 found, 0 missing, 7 empty, 0 corrupt: 100% 286/286 [01:14<00:00,  3.82it/s]\n",
            "\u001b[34m\u001b[1mval: \u001b[0mNew cache created: /content/drive/.shortcut-targets-by-id/1-CXP81xST-t3P2hLZQLjfipgHlqh7k86/yolo-distillation/data/dataset/labels/val.cache\n",
            "Plotting labels to runs/train/yolov5n-distilled2/labels.jpg... \n",
            "\n",
            "\u001b[34m\u001b[1mAutoAnchor: \u001b[0m3.64 anchors/target, 1.000 Best Possible Recall (BPR). Current anchors are a good fit to dataset ✅\n",
            "Image sizes 640 train, 640 val\n",
            "Using 2 dataloader workers\n",
            "Logging results to \u001b[1mruns/train/yolov5n-distilled2\u001b[0m\n",
            "Distillation loss type: l2\n",
            "Starting training for 100 epochs...\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "      0/99     3.86G   0.02644   0.01809  0.008401     23.24        44       640: 100% 62/62 [00:48<00:00,  1.28it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  1.91it/s]\n",
            "                 all        286        396      0.679      0.727      0.732      0.454\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "      1/99      4.2G   0.02736   0.01895   0.00885     18.28        53       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.47it/s]\n",
            "                 all        286        396      0.704      0.668       0.72      0.435\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "      2/99      4.2G    0.0284   0.01904  0.009494     15.38        59       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.71it/s]\n",
            "                 all        286        396      0.689      0.719      0.737      0.441\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "      3/99      4.2G   0.02991   0.01903  0.008823     13.73        49       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.51it/s]\n",
            "                 all        286        396      0.722      0.646      0.724      0.446\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "      4/99      4.2G   0.03045   0.01905  0.009586     12.63        47       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.50it/s]\n",
            "                 all        286        396      0.784      0.658      0.742      0.443\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "      5/99     4.55G   0.03183   0.01913   0.01061     11.83        46       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.43it/s]\n",
            "                 all        286        396      0.797      0.628      0.719       0.43\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "      6/99     4.58G    0.0314   0.01956   0.01047     11.08        49       640: 100% 62/62 [00:41<00:00,  1.50it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.29it/s]\n",
            "                 all        286        396      0.745      0.662       0.73      0.424\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "      7/99     4.58G   0.03288   0.01931   0.01061      10.6        61       640: 100% 62/62 [00:40<00:00,  1.52it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.12it/s]\n",
            "                 all        286        396      0.674      0.625      0.676      0.357\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "      8/99     4.58G   0.03608   0.02003   0.01153     11.12        58       640: 100% 62/62 [00:40<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.71it/s]\n",
            "                 all        286        396      0.602      0.538      0.555      0.303\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "      9/99      4.9G   0.03696   0.01976   0.01088     10.11        46       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.75it/s]\n",
            "                 all        286        396      0.667      0.635      0.652       0.34\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     10/99     4.93G   0.03741   0.01981   0.01089     9.863        54       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.65it/s]\n",
            "                 all        286        396      0.645      0.578      0.597      0.318\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     11/99     4.93G   0.03781   0.02045    0.0116     9.542        54       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.65it/s]\n",
            "                 all        286        396      0.605      0.611      0.627      0.332\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     12/99     4.93G   0.03674   0.01972   0.01139     9.514        37       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.54it/s]\n",
            "                 all        286        396       0.58       0.64      0.623      0.337\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     13/99     4.93G    0.0425    0.0207   0.01229      10.2        46       640: 100% 62/62 [00:37<00:00,  1.63it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.56it/s]\n",
            "                 all        286        396      0.501      0.457      0.436      0.188\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     14/99     4.93G   0.04143   0.02255   0.01566     11.18        39       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.69it/s]\n",
            "                 all        286        396       0.21       0.33      0.203     0.0895\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     15/99     4.93G   0.04247   0.02219   0.01765     10.38        51       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:05<00:00,  1.59it/s]\n",
            "                 all        286        396      0.594      0.513      0.532      0.239\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     16/99     4.93G   0.04012   0.02092   0.01502     9.696        48       640: 100% 62/62 [00:40<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.26it/s]\n",
            "                 all        286        396       0.47      0.553      0.467      0.207\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     17/99     4.93G   0.04084   0.02211   0.01661     9.392        45       640: 100% 62/62 [00:40<00:00,  1.53it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.39it/s]\n",
            "                 all        286        396      0.566      0.506      0.537      0.271\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     18/99     4.93G   0.04081   0.02116   0.01504       8.9        46       640: 100% 62/62 [00:38<00:00,  1.63it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.04it/s]\n",
            "                 all        286        396      0.534      0.504      0.493      0.247\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     19/99     4.93G   0.03822   0.02105   0.01346      8.52        51       640: 100% 62/62 [00:37<00:00,  1.64it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.55it/s]\n",
            "                 all        286        396      0.692      0.587      0.592      0.327\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     20/99     4.93G   0.03963    0.0203    0.0129     8.446        59       640: 100% 62/62 [00:37<00:00,  1.64it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.57it/s]\n",
            "                 all        286        396      0.633       0.58      0.618      0.321\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     21/99     4.93G   0.03779   0.02044   0.01245     8.147        58       640: 100% 62/62 [00:37<00:00,  1.65it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.37it/s]\n",
            "                 all        286        396      0.704      0.614      0.658      0.357\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     22/99     4.93G   0.03758   0.02084   0.01072      7.99        45       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.47it/s]\n",
            "                 all        286        396      0.672      0.623      0.645       0.34\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     23/99     4.93G   0.03641    0.0202   0.01131     7.853        33       640: 100% 62/62 [00:38<00:00,  1.63it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.67it/s]\n",
            "                 all        286        396      0.636      0.652      0.649      0.358\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     24/99     4.93G   0.03759   0.02066    0.0105     7.677        47       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.73it/s]\n",
            "                 all        286        396      0.705      0.662      0.682      0.384\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     25/99     4.93G   0.03663   0.01948   0.01075     7.652        45       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.23it/s]\n",
            "                 all        286        396      0.669      0.622      0.647      0.348\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     26/99     4.93G   0.03495   0.01968   0.01013     7.454        55       640: 100% 62/62 [00:40<00:00,  1.52it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.63it/s]\n",
            "                 all        286        396       0.63      0.563       0.58      0.324\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     27/99     4.93G   0.03562   0.01981   0.01006     7.357        47       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.61it/s]\n",
            "                 all        286        396      0.733      0.623       0.68      0.377\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     28/99     4.93G   0.03541   0.01948   0.01013     7.382        39       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.39it/s]\n",
            "                 all        286        396      0.697      0.647      0.684       0.38\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     29/99     4.93G   0.03612   0.01955   0.01125      7.31        48       640: 100% 62/62 [00:38<00:00,  1.63it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.52it/s]\n",
            "                 all        286        396      0.739       0.69      0.702      0.397\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     30/99     4.93G   0.03464   0.01987   0.01056     7.222        45       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.60it/s]\n",
            "                 all        286        396      0.758      0.682      0.762      0.425\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     31/99     4.93G   0.03315   0.01936  0.009547     7.138        44       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.65it/s]\n",
            "                 all        286        396      0.715      0.622      0.687      0.377\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     32/99     4.93G   0.03485   0.01944  0.009296     6.982        50       640: 100% 62/62 [00:37<00:00,  1.65it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.46it/s]\n",
            "                 all        286        396      0.793      0.632      0.725      0.412\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     33/99     4.93G   0.03423   0.01958   0.01003     6.965        48       640: 100% 62/62 [00:37<00:00,  1.65it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.40it/s]\n",
            "                 all        286        396       0.72      0.644      0.695      0.397\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     34/99     4.93G   0.03357   0.01849  0.009543     6.878        40       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.33it/s]\n",
            "                 all        286        396      0.682      0.648      0.682      0.394\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     35/99     4.93G    0.0335   0.01874  0.009752     6.839        64       640: 100% 62/62 [00:38<00:00,  1.63it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.62it/s]\n",
            "                 all        286        396      0.813      0.663      0.752      0.437\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     36/99     4.93G   0.03345   0.01903  0.009595     6.771        49       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.21it/s]\n",
            "                 all        286        396      0.754      0.632      0.704      0.405\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     37/99     4.93G   0.03262    0.0184  0.009973     6.828        44       640: 100% 62/62 [00:42<00:00,  1.46it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.36it/s]\n",
            "                 all        286        396      0.739      0.623      0.686      0.376\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     38/99     4.93G   0.03335   0.01934  0.008211     6.687        53       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.70it/s]\n",
            "                 all        286        396      0.664      0.707      0.707      0.425\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     39/99     4.93G   0.03254   0.01843  0.008848     6.735        52       640: 100% 62/62 [00:37<00:00,  1.64it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.40it/s]\n",
            "                 all        286        396      0.806      0.644      0.749      0.437\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     40/99     4.93G   0.03272   0.01863  0.008718     6.618        45       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.74it/s]\n",
            "                 all        286        396       0.72      0.694      0.729      0.425\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     41/99     4.93G   0.03224   0.01851  0.008047     6.551        31       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.49it/s]\n",
            "                 all        286        396      0.737      0.729      0.754      0.453\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     42/99     4.93G    0.0326   0.01881  0.008838     6.535        63       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.39it/s]\n",
            "                 all        286        396      0.677       0.72      0.734      0.416\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     43/99     4.93G    0.0317   0.01842  0.007914     6.512        53       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.98it/s]\n",
            "                 all        286        396      0.649      0.662       0.68      0.403\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     44/99     4.93G   0.03205   0.01856  0.008354      6.45        44       640: 100% 62/62 [00:40<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.74it/s]\n",
            "                 all        286        396      0.761      0.652      0.718      0.436\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     45/99     4.93G   0.03164   0.01858  0.008223     6.443        43       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.25it/s]\n",
            "                 all        286        396      0.757      0.679      0.729      0.429\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     46/99     4.93G   0.03253   0.01785   0.00966     6.501        45       640: 100% 62/62 [00:37<00:00,  1.64it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.61it/s]\n",
            "                 all        286        396      0.714      0.729      0.748      0.438\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     47/99     4.93G   0.03177   0.01877  0.006832       6.4        50       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.89it/s]\n",
            "                 all        286        396      0.768      0.694      0.751      0.432\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     48/99     4.93G   0.03134   0.01815  0.007945     6.357        42       640: 100% 62/62 [00:37<00:00,  1.64it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.43it/s]\n",
            "                 all        286        396      0.766      0.676       0.74      0.436\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     49/99     4.93G   0.03079   0.01784  0.008904     6.393        50       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.71it/s]\n",
            "                 all        286        396      0.731       0.75      0.768      0.454\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     50/99     4.93G   0.03136   0.01834  0.007846     6.286        52       640: 100% 62/62 [00:41<00:00,  1.51it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.38it/s]\n",
            "                 all        286        396       0.78      0.657      0.716      0.432\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     51/99     4.93G   0.03073   0.01777  0.007856     6.252        40       640: 100% 62/62 [00:40<00:00,  1.53it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396      0.804      0.742       0.78      0.466\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     52/99     4.93G     0.031   0.01782  0.008676      6.27        39       640: 100% 62/62 [00:41<00:00,  1.48it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.61it/s]\n",
            "                 all        286        396       0.67      0.758      0.727      0.445\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     53/99     4.93G   0.03147   0.01791  0.007882     6.138        36       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.57it/s]\n",
            "                 all        286        396      0.776      0.693      0.748      0.462\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     54/99     4.93G   0.02983   0.01877   0.00726     6.142        41       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.59it/s]\n",
            "                 all        286        396      0.751      0.718      0.753      0.459\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     55/99     4.93G   0.02983    0.0173  0.006332     6.163        51       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.38it/s]\n",
            "                 all        286        396       0.72      0.768      0.784      0.488\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     56/99     4.93G   0.02932   0.01753  0.007035     6.093        35       640: 100% 62/62 [00:40<00:00,  1.51it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.55it/s]\n",
            "                 all        286        396      0.815      0.672       0.77      0.472\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     57/99     4.93G   0.02886   0.01779  0.007667      6.08        55       640: 100% 62/62 [00:37<00:00,  1.64it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.47it/s]\n",
            "                 all        286        396      0.744      0.774      0.768      0.475\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     58/99     4.93G   0.03013    0.0181  0.006777     6.024        30       640: 100% 62/62 [00:38<00:00,  1.63it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.40it/s]\n",
            "                 all        286        396      0.784       0.74      0.781       0.49\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     59/99     4.93G   0.02968   0.01778  0.007103      6.08        45       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.13it/s]\n",
            "                 all        286        396      0.778       0.75      0.774      0.479\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     60/99     4.93G   0.03033    0.0181  0.005887     6.045        57       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.27it/s]\n",
            "                 all        286        396      0.775      0.729       0.78      0.477\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     61/99     4.93G    0.0287   0.01728  0.005902     5.977        39       640: 100% 62/62 [00:40<00:00,  1.53it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.19it/s]\n",
            "                 all        286        396      0.833      0.683      0.774      0.473\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     62/99     4.93G   0.02948   0.01787   0.00694     5.983        45       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  1.99it/s]\n",
            "                 all        286        396      0.784      0.715      0.769      0.485\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     63/99     4.93G   0.02962   0.01804  0.007111     5.966        40       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.52it/s]\n",
            "                 all        286        396       0.73      0.711      0.762      0.469\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     64/99     4.93G   0.02839   0.01818  0.006318     5.892        54       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.41it/s]\n",
            "                 all        286        396      0.808        0.7      0.785      0.492\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     65/99     4.93G   0.02866   0.01738  0.006382     5.931        38       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.84it/s]\n",
            "                 all        286        396      0.787      0.732      0.784      0.502\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     66/99     4.93G   0.02879   0.01775  0.006937     5.932        44       640: 100% 62/62 [00:41<00:00,  1.51it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.45it/s]\n",
            "                 all        286        396      0.769      0.726      0.789      0.492\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     67/99     4.93G   0.02831   0.01763  0.006708     5.882        50       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.60it/s]\n",
            "                 all        286        396      0.784      0.727      0.791      0.501\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     68/99     4.93G   0.02811   0.01682  0.006384      5.81        42       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.36it/s]\n",
            "                 all        286        396      0.812      0.725      0.791      0.486\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     69/99     4.93G   0.02763   0.01755  0.006036     5.842        48       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.40it/s]\n",
            "                 all        286        396      0.759      0.759      0.797      0.514\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     70/99     4.93G    0.0279   0.01702  0.007096     5.855        40       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396      0.767      0.766      0.796      0.503\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     71/99     4.93G   0.02853   0.01716  0.006779     5.806        41       640: 100% 62/62 [00:38<00:00,  1.62it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.51it/s]\n",
            "                 all        286        396      0.743      0.738      0.774      0.488\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     72/99     4.93G   0.02699   0.01721  0.006102     5.785        56       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.61it/s]\n",
            "                 all        286        396      0.821      0.724      0.793      0.504\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     73/99     4.93G   0.02737   0.01722  0.006149     5.759        57       640: 100% 62/62 [00:37<00:00,  1.64it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.49it/s]\n",
            "                 all        286        396       0.82      0.708      0.782      0.488\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     74/99     4.93G   0.02698   0.01705  0.005555     5.722        44       640: 100% 62/62 [00:37<00:00,  1.64it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.45it/s]\n",
            "                 all        286        396      0.768      0.755      0.796      0.507\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     75/99     4.93G   0.02664   0.01671  0.006327     5.782        47       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.64it/s]\n",
            "                 all        286        396      0.756      0.773      0.783      0.497\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     76/99     4.93G   0.02684   0.01777  0.005942      5.68        43       640: 100% 62/62 [00:37<00:00,  1.63it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.61it/s]\n",
            "                 all        286        396      0.823      0.739      0.798       0.51\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     77/99     4.93G   0.02649   0.01752  0.005849     5.707        46       640: 100% 62/62 [00:37<00:00,  1.66it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.38it/s]\n",
            "                 all        286        396      0.815      0.727      0.802      0.518\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     78/99     4.93G   0.02705   0.01692  0.006062     5.739        35       640: 100% 62/62 [00:40<00:00,  1.52it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.62it/s]\n",
            "                 all        286        396       0.77      0.743      0.799      0.517\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     79/99     4.93G    0.0267    0.0169  0.006238     5.639        44       640: 100% 62/62 [00:37<00:00,  1.65it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.46it/s]\n",
            "                 all        286        396      0.759       0.76      0.801      0.519\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     80/99     4.93G   0.02696   0.01692  0.006631     5.714        44       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.68it/s]\n",
            "                 all        286        396      0.817      0.724      0.794      0.513\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     81/99     4.93G   0.02584   0.01667  0.005536     5.666        55       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.51it/s]\n",
            "                 all        286        396      0.832      0.713      0.795      0.509\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     82/99     4.93G   0.02641   0.01653  0.006051     5.651        36       640: 100% 62/62 [00:37<00:00,  1.63it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.52it/s]\n",
            "                 all        286        396      0.791      0.771      0.815      0.521\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     83/99     4.93G   0.02623   0.01679  0.006424     5.598        41       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.40it/s]\n",
            "                 all        286        396      0.838        0.7      0.792      0.513\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     84/99     4.93G   0.02517   0.01674  0.004837     5.626        49       640: 100% 62/62 [00:37<00:00,  1.65it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  1.89it/s]\n",
            "                 all        286        396      0.757      0.785      0.802      0.524\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     85/99     4.93G   0.02637   0.01716  0.005724     5.569        39       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.43it/s]\n",
            "                 all        286        396      0.812      0.747      0.812      0.518\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     86/99     4.93G   0.02636   0.01662  0.005873     5.623        51       640: 100% 62/62 [00:37<00:00,  1.65it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.56it/s]\n",
            "                 all        286        396      0.797      0.763      0.807      0.523\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     87/99     4.93G   0.02599   0.01639  0.006252     5.565        47       640: 100% 62/62 [00:37<00:00,  1.63it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.68it/s]\n",
            "                 all        286        396      0.785      0.747        0.8      0.519\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     88/99     4.93G   0.02591   0.01581  0.004681      5.57        55       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.44it/s]\n",
            "                 all        286        396      0.793       0.77      0.813      0.528\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     89/99     4.93G     0.026   0.01646  0.005126     5.548        45       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.48it/s]\n",
            "                 all        286        396      0.767      0.784      0.812       0.52\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     90/99     4.93G   0.02508   0.01596  0.005459     5.549        40       640: 100% 62/62 [00:38<00:00,  1.63it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.43it/s]\n",
            "                 all        286        396      0.797      0.756       0.81      0.532\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     91/99     4.93G   0.02527   0.01611  0.004956     5.526        43       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.71it/s]\n",
            "                 all        286        396      0.798      0.759      0.807      0.524\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     92/99     4.93G   0.02547   0.01651  0.005052     5.556        37       640: 100% 62/62 [00:37<00:00,  1.64it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.49it/s]\n",
            "                 all        286        396      0.767      0.784      0.797      0.527\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     93/99     4.93G   0.02507   0.01647  0.005127     5.506        56       640: 100% 62/62 [00:37<00:00,  1.65it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.53it/s]\n",
            "                 all        286        396      0.828      0.755      0.806      0.524\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     94/99     4.93G   0.02485   0.01618  0.005083     5.532        41       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:02<00:00,  3.13it/s]\n",
            "                 all        286        396      0.718      0.811      0.801      0.522\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     95/99     4.93G   0.02479   0.01604  0.005638     5.505        50       640: 100% 62/62 [00:37<00:00,  1.65it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.40it/s]\n",
            "                 all        286        396      0.834      0.711        0.8      0.527\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     96/99     4.93G   0.02498   0.01685  0.005272     5.456        38       640: 100% 62/62 [00:37<00:00,  1.66it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.50it/s]\n",
            "                 all        286        396      0.788      0.764      0.803      0.526\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     97/99     4.93G   0.02429   0.01618  0.004651     5.489        51       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.51it/s]\n",
            "                 all        286        396      0.828      0.738      0.806      0.531\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     98/99     4.93G   0.02507   0.01626  0.005786     5.488        48       640: 100% 62/62 [00:37<00:00,  1.65it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.57it/s]\n",
            "                 all        286        396      0.793      0.756      0.813      0.529\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls   feature    labels  img_size\n",
            "     99/99     4.93G   0.02534   0.01667  0.005391     5.457        37       640: 100% 62/62 [00:37<00:00,  1.66it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.55it/s]\n",
            "                 all        286        396      0.827      0.732      0.811      0.531\n",
            "\n",
            "100 epochs completed in 1.257 hours.\n",
            "Optimizer stripped from runs/train/yolov5n-distilled2/weights/last.pt, 275.7MB\n",
            "Optimizer stripped from runs/train/yolov5n-distilled2/weights/best.pt, 275.7MB\n",
            "\n",
            "Validating runs/train/yolov5n-distilled2/weights/best.pt...\n",
            "Fusing layers... \n",
            "Model Summary: 232 layers, 26887978 parameters, 0 gradients, 4.1 GFLOPs\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  1.81it/s]\n",
            "                 all        286        396      0.824      0.736      0.808      0.532\n",
            "                 cat        286         82      0.822      0.793      0.824      0.511\n",
            "                 cow        286         44      0.746      0.818      0.842      0.622\n",
            "               sheep        286         60      0.912      0.515      0.716      0.447\n",
            "                 dog        286        122       0.75      0.713      0.754      0.467\n",
            "               horse        286         88      0.891      0.841      0.906      0.614\n",
            "Results saved to \u001b[1mruns/train/yolov5n-distilled2\u001b[0m\n"
          ]
        }
      ],
      "source": [
        "!python train_dist.py --weights weights/yolov5n_best.pt --cfg models/yolov5n_animal.yaml --data data/VOC_animal.yaml --batch-size 16 --epochs 100 --workers 2 --t_weights weights/yolov5l_best.pt --hyp data/hyps/hyp.scratch-low-distillation.yaml --name yolov5n-distilled --distill_cfg configs/distill_config2.py"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "!python train_dist.py --weights weights/yolov5n_best.pt --cfg models/yolov5n_animal.yaml --data data/VOC_animal.yaml --batch-size 16 --epochs 100 --workers 2 --t_weights weights/yolov5l_best.pt --hyp data/hyps/hyp.scratch-low-distillation.yaml --name yolov5n-distilled --distill_cfg configs/distill_config2.py"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "8vZ9zsocpItp",
        "outputId": "0336e464-8562-49b9-bb7a-7062b9637f7e"
      },
      "execution_count": 8,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Generating grammar tables from /usr/lib/python3.7/lib2to3/Grammar.txt\n",
            "Generating grammar tables from /usr/lib/python3.7/lib2to3/PatternGrammar.txt\n",
            "\u001b[34m\u001b[1mtrain_dist: \u001b[0mweights=weights/yolov5n_best.pt, t_weights=weights/yolov5l_best.pt, dist_loss=l2, temperature=20, distill_cfg=configs/distill_config2.py, cfg=models/yolov5n_animal.yaml, data=data/VOC_animal.yaml, hyp=data/hyps/hyp.scratch-low-distillation.yaml, epochs=100, batch_size=16, imgsz=640, rect=False, resume=False, nosave=False, noval=False, noautoanchor=False, evolve=None, bucket=, cache=None, image_weights=False, device=, multi_scale=False, single_cls=False, optimizer=SGD, sync_bn=False, workers=2, project=runs/train, name=yolov5n-distilled, exist_ok=False, quad=False, cos_lr=False, label_smoothing=0.0, patience=100, freeze=[0], save_period=-1, local_rank=-1, entity=None, upload_dataset=False, bbox_interval=-1, artifact_alias=latest\n",
            "\u001b[34m\u001b[1mgithub: \u001b[0mup to date with https://gitee.com/zfhwang/yolo-distillation ✅\n",
            "YOLOv5 🚀 3f3185b torch 1.12.1+cu113 CUDA:0 (Tesla T4, 15110MiB)\n",
            "\n",
            "\u001b[34m\u001b[1mhyperparameters: \u001b[0mlr0=0.01, lrf=0.01, momentum=0.9, weight_decay=0.0001, warmup_epochs=3.0, warmup_momentum=0.8, warmup_bias_lr=0.1, box=0.05, cls=0.5, cls_pw=1.0, obj=1.0, obj_pw=1.0, iou_t=0.2, anchor_t=4.0, fl_gamma=0.0, hsv_h=0.015, hsv_s=0.7, hsv_v=0.4, degrees=0.0, translate=0.1, scale=0.5, shear=0.0, perspective=0.0, flipud=0.0, fliplr=0.5, mosaic=1.0, mixup=0.0, copy_paste=0.0, dist=0.5\n",
            "\u001b[34m\u001b[1mWeights & Biases: \u001b[0mrun 'pip install wandb' to automatically track and visualize YOLOv5 🚀 runs (RECOMMENDED)\n",
            "\u001b[34m\u001b[1mTensorBoard: \u001b[0mStart with 'tensorboard --logdir runs/train', view at http://localhost:6006/\n",
            "\n",
            "                 from  n    params  module                                  arguments                     \n",
            "  0                -1  1      1760  models.common.Conv                      [3, 16, 6, 2, 2]              \n",
            "  1                -1  1      4672  models.common.Conv                      [16, 32, 3, 2]                \n",
            "  2                -1  1      4800  models.common.C3                        [32, 32, 1]                   \n",
            "  3                -1  1     18560  models.common.Conv                      [32, 64, 3, 2]                \n",
            "  4                -1  2     29184  models.common.C3                        [64, 64, 2]                   \n",
            "  5                -1  1     73984  models.common.Conv                      [64, 128, 3, 2]               \n",
            "  6                -1  3    156928  models.common.C3                        [128, 128, 3]                 \n",
            "  7                -1  1    295424  models.common.Conv                      [128, 256, 3, 2]              \n",
            "  8                -1  1    296448  models.common.C3                        [256, 256, 1]                 \n",
            "  9                -1  1    164608  models.common.SPPF                      [256, 256, 5]                 \n",
            " 10                -1  1     33024  models.common.Conv                      [256, 128, 1, 1]              \n",
            " 11                -1  1         0  torch.nn.modules.upsampling.Upsample    [None, 2, 'nearest']          \n",
            " 12           [-1, 6]  1         0  models.common.Concat                    [1]                           \n",
            " 13                -1  1     90880  models.common.C3                        [256, 128, 1, False]          \n",
            " 14                -1  1      8320  models.common.Conv                      [128, 64, 1, 1]               \n",
            " 15                -1  1         0  torch.nn.modules.upsampling.Upsample    [None, 2, 'nearest']          \n",
            " 16           [-1, 4]  1         0  models.common.Concat                    [1]                           \n",
            " 17                -1  1     22912  models.common.C3                        [128, 64, 1, False]           \n",
            " 18                -1  1     36992  models.common.Conv                      [64, 64, 3, 2]                \n",
            " 19          [-1, 14]  1         0  models.common.Concat                    [1]                           \n",
            " 20                -1  1     74496  models.common.C3                        [128, 128, 1, False]          \n",
            " 21                -1  1    147712  models.common.Conv                      [128, 128, 3, 2]              \n",
            " 22          [-1, 10]  1         0  models.common.Concat                    [1]                           \n",
            " 23                -1  1    296448  models.common.C3                        [256, 256, 1, False]          \n",
            " 24      [17, 20, 23]  1     13530  models.yolo.Detect                      [5, [[10, 13, 16, 30, 33, 23], [30, 61, 62, 45, 59, 119], [116, 90, 156, 198, 373, 326]], [64, 128, 256]]\n",
            "Model Summary: 270 layers, 1770682 parameters, 1770682 gradients, 4.2 GFLOPs\n",
            "\n",
            "Transferred 348/349 items from weights/yolov5n_best.pt\n",
            "load t-model from weights/yolov5l_best.pt\n",
            "Scaled weight_decay = 0.0001\n",
            "\u001b[34m\u001b[1moptimizer:\u001b[0m SGD with parameter groups 57 weight (no decay), 69 weight, 69 bias\n",
            "\u001b[34m\u001b[1malbumentations: \u001b[0mBlur(always_apply=False, p=0.01, blur_limit=(3, 7)), MedianBlur(always_apply=False, p=0.01, blur_limit=(3, 7)), ToGray(always_apply=False, p=0.01), CLAHE(always_apply=False, p=0.01, clip_limit=(1, 4.0), tile_grid_size=(8, 8))\n",
            "\u001b[34m\u001b[1mtrain: \u001b[0mScanning '/content/drive/.shortcut-targets-by-id/1-CXP81xST-t3P2hLZQLjfipgHlqh7k86/yolo-distillation/data/dataset/labels/train.cache' images and labels... 991 found, 0 missing, 16 empty, 0 corrupt: 100% 991/991 [00:00<?, ?it/s]\n",
            "\u001b[34m\u001b[1mval: \u001b[0mScanning '/content/drive/.shortcut-targets-by-id/1-CXP81xST-t3P2hLZQLjfipgHlqh7k86/yolo-distillation/data/dataset/labels/val.cache' images and labels... 286 found, 0 missing, 7 empty, 0 corrupt: 100% 286/286 [00:00<?, ?it/s]\n",
            "Plotting labels to runs/train/yolov5n-distilled4/labels.jpg... \n",
            "\n",
            "\u001b[34m\u001b[1mAutoAnchor: \u001b[0m3.64 anchors/target, 1.000 Best Possible Recall (BPR). Current anchors are a good fit to dataset ✅\n",
            "Image sizes 640 train, 640 val\n",
            "Using 2 dataloader workers\n",
            "Logging results to \u001b[1mruns/train/yolov5n-distilled4\u001b[0m\n",
            "Distillation loss type: l2\n",
            "Starting training for 100 epochs...\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      0/99     3.82G   0.02653   0.02095    0.0084        44       640: 100% 62/62 [00:43<00:00,  1.42it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.60it/s]\n",
            "                 all        286        396      0.727      0.658      0.731      0.451\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      1/99     4.17G   0.02732   0.02287  0.008728        53       640: 100% 62/62 [00:40<00:00,  1.53it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.36it/s]\n",
            "                 all        286        396      0.739      0.629      0.705      0.427\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      2/99     4.17G    0.0283   0.02339  0.009467        59       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.29it/s]\n",
            "                 all        286        396      0.695       0.66      0.697      0.411\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      3/99     4.17G   0.02974   0.02314  0.008617        49       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.35it/s]\n",
            "                 all        286        396      0.691      0.657      0.698      0.416\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      4/99     4.17G   0.02999    0.0235  0.009001        47       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.36it/s]\n",
            "                 all        286        396      0.724      0.611      0.703      0.397\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      5/99     4.52G   0.03086   0.02357  0.009808        46       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.35it/s]\n",
            "                 all        286        396      0.732      0.632      0.693      0.407\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      6/99     4.54G   0.03134   0.02444   0.01016        49       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.32it/s]\n",
            "                 all        286        396      0.659      0.722      0.731      0.412\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      7/99     4.54G    0.0325    0.0248   0.01071        61       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.30it/s]\n",
            "                 all        286        396      0.716      0.508       0.63      0.319\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      8/99     4.54G   0.03592   0.02467   0.01015        58       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.55it/s]\n",
            "                 all        286        396      0.635      0.598      0.623      0.334\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "      9/99     4.87G   0.03646   0.02607   0.01032        46       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.40it/s]\n",
            "                 all        286        396      0.611      0.569      0.585      0.314\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     10/99     4.89G   0.03706   0.02589   0.01085        54       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.49it/s]\n",
            "                 all        286        396      0.578      0.605      0.579      0.304\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     11/99     4.89G   0.03913   0.02703   0.01135        54       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.30it/s]\n",
            "                 all        286        396      0.627      0.551      0.591       0.29\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     12/99     4.89G   0.03887   0.03039   0.01382        37       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.24it/s]\n",
            "                 all        286        396      0.522      0.276      0.262       0.11\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     13/99     4.89G   0.04085   0.03073   0.01508        46       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.73it/s]\n",
            "                 all        286        396      0.556      0.373      0.379      0.156\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     14/99     4.89G   0.04316   0.03152   0.01565        39       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.20it/s]\n",
            "                 all        286        396      0.588      0.336      0.375      0.173\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     15/99     4.89G   0.04461   0.03173   0.01706        51       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.48it/s]\n",
            "                 all        286        396      0.487      0.357      0.347      0.149\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     16/99     4.89G   0.04158   0.03013   0.01543        48       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.31it/s]\n",
            "                 all        286        396      0.544      0.339      0.312      0.119\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     17/99     4.89G   0.04065   0.02945   0.01481        45       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.32it/s]\n",
            "                 all        286        396      0.559      0.507      0.503      0.243\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     18/99     4.89G   0.04089   0.02933   0.01499        46       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.60it/s]\n",
            "                 all        286        396      0.533       0.55      0.525      0.254\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     19/99     4.89G      0.04   0.02813   0.01243        51       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.45it/s]\n",
            "                 all        286        396       0.63      0.614       0.64      0.308\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     20/99     4.89G   0.03969   0.02619   0.01153        59       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.40it/s]\n",
            "                 all        286        396      0.618      0.603       0.64      0.295\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     21/99     4.89G    0.0402   0.02638   0.01181        58       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.32it/s]\n",
            "                 all        286        396      0.644      0.552      0.614      0.307\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     22/99     4.89G   0.03928   0.02738   0.01033        45       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.63it/s]\n",
            "                 all        286        396      0.624      0.586       0.61      0.306\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     23/99     4.89G   0.03825   0.02558     0.011        33       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.33it/s]\n",
            "                 all        286        396      0.699      0.545      0.632      0.327\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     24/99     4.89G   0.03862    0.0261   0.01059        47       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.51it/s]\n",
            "                 all        286        396      0.673      0.626      0.652      0.346\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     25/99     4.89G   0.03828   0.02445  0.009627        45       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.29it/s]\n",
            "                 all        286        396      0.695      0.626      0.655      0.352\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     26/99     4.89G    0.0357   0.02419   0.01025        55       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.35it/s]\n",
            "                 all        286        396      0.644      0.594      0.642      0.336\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     27/99     4.89G   0.03593   0.02413  0.009493        47       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.33it/s]\n",
            "                 all        286        396      0.684      0.645      0.672      0.385\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     28/99     4.89G   0.03468   0.02417      0.01        39       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.31it/s]\n",
            "                 all        286        396      0.686      0.651      0.692      0.369\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     29/99     4.89G   0.03641   0.02428   0.01015        48       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.35it/s]\n",
            "                 all        286        396      0.704      0.611      0.692      0.378\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     30/99     4.89G   0.03503   0.02343   0.01032        45       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.35it/s]\n",
            "                 all        286        396      0.734      0.662       0.74      0.407\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     31/99     4.89G   0.03397   0.02287    0.0088        44       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.50it/s]\n",
            "                 all        286        396      0.683      0.654      0.677      0.388\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     32/99     4.89G   0.03519   0.02277   0.00858        50       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.44it/s]\n",
            "                 all        286        396       0.72      0.655      0.714      0.397\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     33/99     4.89G   0.03393    0.0228  0.009495        48       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.58it/s]\n",
            "                 all        286        396      0.731      0.611      0.685      0.397\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     34/99     4.89G   0.03354   0.02242  0.009844        40       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.46it/s]\n",
            "                 all        286        396      0.606       0.53      0.592      0.329\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     35/99     4.89G   0.03376   0.02204   0.00968        64       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396      0.695      0.651      0.683      0.368\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     36/99     4.89G   0.03361   0.02237  0.009133        49       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.47it/s]\n",
            "                 all        286        396      0.798      0.634      0.714      0.412\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     37/99     4.89G    0.0321   0.02131  0.009453        44       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.31it/s]\n",
            "                 all        286        396      0.714       0.64      0.689      0.379\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     38/99     4.89G   0.03337   0.02214  0.008129        53       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396      0.731      0.685      0.725      0.425\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     39/99     4.89G   0.03325   0.02139  0.008583        52       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396      0.823      0.665      0.767      0.441\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     40/99     4.89G   0.03312   0.02159  0.008562        45       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.81it/s]\n",
            "                 all        286        396      0.708      0.731      0.741      0.432\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     41/99     4.89G   0.03184   0.02122  0.008341        31       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.43it/s]\n",
            "                 all        286        396      0.718      0.688      0.711      0.417\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     42/99     4.89G   0.03309   0.02142  0.008483        63       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.37it/s]\n",
            "                 all        286        396       0.69      0.716       0.74        0.4\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     43/99     4.89G   0.03185   0.02099  0.007607        53       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.52it/s]\n",
            "                 all        286        396      0.644      0.616      0.644       0.37\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     44/99     4.89G   0.03145   0.02112  0.007757        44       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.52it/s]\n",
            "                 all        286        396      0.798      0.645       0.73      0.426\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     45/99     4.89G   0.03125   0.02078  0.007882        43       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.45it/s]\n",
            "                 all        286        396      0.797      0.672      0.741      0.437\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     46/99     4.89G   0.03241   0.01978  0.008843        45       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.40it/s]\n",
            "                 all        286        396      0.754      0.678       0.74      0.433\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     47/99     4.89G   0.03145   0.02081  0.006935        50       640: 100% 62/62 [00:38<00:00,  1.61it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.37it/s]\n",
            "                 all        286        396      0.746      0.706      0.758      0.449\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     48/99     4.89G   0.03107   0.02029   0.00761        42       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.32it/s]\n",
            "                 all        286        396      0.706      0.722      0.727      0.426\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     49/99     4.89G   0.03071   0.01984  0.008188        50       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396      0.758      0.695      0.751       0.44\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     50/99     4.89G   0.03085   0.02019  0.007318        52       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.59it/s]\n",
            "                 all        286        396      0.788      0.662      0.747      0.442\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     51/99     4.89G   0.03008   0.01967  0.007912        40       640: 100% 62/62 [00:39<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.31it/s]\n",
            "                 all        286        396      0.759      0.721       0.76      0.446\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     52/99     4.89G   0.03073   0.01968  0.008518        39       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.38it/s]\n",
            "                 all        286        396      0.679      0.696      0.724      0.438\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     53/99     4.89G   0.03051   0.01954  0.007596        36       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.73it/s]\n",
            "                 all        286        396      0.771      0.692      0.751       0.46\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     54/99     4.89G   0.02931   0.02024   0.00693        41       640: 100% 62/62 [00:40<00:00,  1.53it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.39it/s]\n",
            "                 all        286        396      0.773      0.657      0.747      0.437\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     55/99     4.89G   0.02902   0.01877  0.006315        51       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.48it/s]\n",
            "                 all        286        396      0.764      0.704       0.77      0.473\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     56/99     4.89G   0.02875   0.01908  0.007034        35       640: 100% 62/62 [00:40<00:00,  1.53it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.44it/s]\n",
            "                 all        286        396      0.819      0.662      0.753      0.455\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     57/99     4.89G   0.02849   0.01926  0.007687        55       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396      0.761      0.721      0.751      0.453\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     58/99     4.89G   0.02929   0.01959  0.006617        30       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.46it/s]\n",
            "                 all        286        396      0.791      0.705      0.772      0.479\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     59/99     4.89G   0.02941   0.01926  0.007039        45       640: 100% 62/62 [00:40<00:00,  1.53it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.47it/s]\n",
            "                 all        286        396      0.752      0.718      0.757      0.466\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     60/99     4.89G   0.02939   0.01938  0.006176        57       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.46it/s]\n",
            "                 all        286        396      0.809      0.712      0.778      0.481\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     61/99     4.89G   0.02851   0.01852  0.005543        39       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.25it/s]\n",
            "                 all        286        396      0.777      0.686      0.771      0.465\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     62/99     4.89G   0.02838   0.01912  0.006768        45       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.40it/s]\n",
            "                 all        286        396      0.763       0.72      0.767      0.475\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     63/99     4.89G   0.02866   0.01901  0.006742        40       640: 100% 62/62 [00:38<00:00,  1.59it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.34it/s]\n",
            "                 all        286        396      0.777       0.68      0.746      0.462\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     64/99     4.89G   0.02788   0.01907  0.006141        54       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.41it/s]\n",
            "                 all        286        396       0.79      0.685      0.775      0.491\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     65/99     4.89G   0.02769   0.01848  0.006233        38       640: 100% 62/62 [00:40<00:00,  1.53it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396      0.839      0.705      0.785      0.486\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     66/99     4.89G   0.02791   0.01872  0.006439        44       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.57it/s]\n",
            "                 all        286        396       0.78      0.713      0.781      0.496\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     67/99     4.89G   0.02709   0.01831  0.006201        50       640: 100% 62/62 [00:40<00:00,  1.52it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.31it/s]\n",
            "                 all        286        396      0.762      0.745      0.769       0.48\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     68/99     4.89G    0.0272   0.01757  0.006219        42       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.34it/s]\n",
            "                 all        286        396      0.774      0.759      0.785      0.483\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     69/99     4.89G   0.02681   0.01841  0.005781        48       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.38it/s]\n",
            "                 all        286        396      0.785      0.744       0.78      0.501\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     70/99     4.89G   0.02701   0.01792  0.007225        40       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.26it/s]\n",
            "                 all        286        396      0.773      0.733      0.791      0.504\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     71/99     4.89G   0.02791   0.01785  0.006817        41       640: 100% 62/62 [00:40<00:00,  1.53it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.58it/s]\n",
            "                 all        286        396      0.742       0.72      0.774      0.488\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     72/99     4.89G   0.02602   0.01787  0.005895        56       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.42it/s]\n",
            "                 all        286        396      0.782      0.767      0.788      0.501\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     73/99     4.89G   0.02676    0.0178  0.005651        57       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.32it/s]\n",
            "                 all        286        396      0.794      0.718      0.781      0.492\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     74/99     4.89G   0.02616   0.01764  0.005583        44       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.46it/s]\n",
            "                 all        286        396      0.777      0.746      0.793      0.515\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     75/99     4.89G   0.02586   0.01717  0.006267        47       640: 100% 62/62 [00:40<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.28it/s]\n",
            "                 all        286        396      0.768      0.722      0.774      0.502\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     76/99     4.89G   0.02561   0.01808  0.006127        43       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.34it/s]\n",
            "                 all        286        396      0.803      0.728      0.799      0.516\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     77/99     4.89G   0.02579   0.01796  0.005679        46       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.27it/s]\n",
            "                 all        286        396      0.783      0.745      0.792      0.503\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     78/99     4.89G   0.02605   0.01756  0.006319        35       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.38it/s]\n",
            "                 all        286        396      0.793      0.731      0.795      0.511\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     79/99     4.89G   0.02558    0.0173  0.006172        44       640: 100% 62/62 [00:38<00:00,  1.60it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.37it/s]\n",
            "                 all        286        396      0.776      0.764      0.802      0.521\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     80/99     4.89G   0.02583   0.01725  0.006455        44       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.48it/s]\n",
            "                 all        286        396      0.785      0.751      0.793      0.505\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     81/99     4.89G   0.02511   0.01721   0.00513        55       640: 100% 62/62 [00:39<00:00,  1.57it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.37it/s]\n",
            "                 all        286        396      0.809      0.738        0.8      0.514\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     82/99     4.89G    0.0257   0.01679  0.005932        36       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.48it/s]\n",
            "                 all        286        396      0.783      0.782      0.811      0.522\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     83/99     4.89G   0.02528   0.01717  0.005754        41       640: 100% 62/62 [00:40<00:00,  1.52it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.48it/s]\n",
            "                 all        286        396      0.806      0.755      0.798      0.517\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     84/99     4.89G   0.02436   0.01714   0.00458        49       640: 100% 62/62 [00:39<00:00,  1.58it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.32it/s]\n",
            "                 all        286        396      0.797      0.753      0.798       0.52\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     85/99     4.89G   0.02561   0.01738  0.005531        39       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.38it/s]\n",
            "                 all        286        396      0.809      0.758      0.805      0.524\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     86/99     4.89G   0.02516   0.01709  0.005917        51       640: 100% 62/62 [00:40<00:00,  1.52it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.49it/s]\n",
            "                 all        286        396      0.812      0.751      0.806      0.522\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     87/99     4.89G   0.02498   0.01669  0.006277        47       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.30it/s]\n",
            "                 all        286        396      0.808      0.753      0.801      0.515\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     88/99     4.89G   0.02518   0.01618  0.004629        55       640: 100% 62/62 [00:41<00:00,  1.48it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.20it/s]\n",
            "                 all        286        396      0.817       0.75      0.809      0.523\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     89/99     4.89G   0.02515   0.01674    0.0051        45       640: 100% 62/62 [00:41<00:00,  1.49it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.15it/s]\n",
            "                 all        286        396      0.763      0.782      0.803      0.513\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     90/99     4.89G   0.02411   0.01611  0.005066        40       640: 100% 62/62 [00:42<00:00,  1.47it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.33it/s]\n",
            "                 all        286        396      0.772      0.767      0.796      0.516\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     91/99     4.89G   0.02432   0.01644  0.004745        43       640: 100% 62/62 [00:41<00:00,  1.50it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.23it/s]\n",
            "                 all        286        396      0.754      0.769      0.798      0.522\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     92/99     4.89G   0.02468   0.01658  0.005086        37       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.41it/s]\n",
            "                 all        286        396      0.792      0.744      0.796      0.517\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     93/99     4.89G   0.02403   0.01653  0.004816        56       640: 100% 62/62 [00:39<00:00,  1.55it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.29it/s]\n",
            "                 all        286        396       0.77      0.758      0.795      0.521\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     94/99     4.89G   0.02373   0.01631  0.004829        41       640: 100% 62/62 [00:39<00:00,  1.56it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.28it/s]\n",
            "                 all        286        396      0.778      0.781      0.795      0.517\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     95/99     4.89G   0.02374    0.0161  0.005284        50       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.27it/s]\n",
            "                 all        286        396      0.798      0.749      0.788      0.514\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     96/99     4.89G   0.02391   0.01689  0.004974        38       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.37it/s]\n",
            "                 all        286        396      0.781       0.77      0.796       0.52\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     97/99     4.89G   0.02322   0.01627  0.004693        51       640: 100% 62/62 [00:40<00:00,  1.53it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.69it/s]\n",
            "                 all        286        396      0.792      0.744      0.796      0.521\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     98/99     4.89G   0.02391   0.01636  0.005584        48       640: 100% 62/62 [00:40<00:00,  1.53it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:03<00:00,  2.56it/s]\n",
            "                 all        286        396      0.786      0.771      0.805      0.521\n",
            "\n",
            "     Epoch   gpu_mem       box       obj       cls    labels  img_size\n",
            "     99/99     4.89G   0.02413   0.01668  0.005049        37       640: 100% 62/62 [00:40<00:00,  1.54it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  2.24it/s]\n",
            "                 all        286        396       0.79      0.772      0.809      0.523\n",
            "\n",
            "100 epochs completed in 1.279 hours.\n",
            "Optimizer stripped from runs/train/yolov5n-distilled4/weights/last.pt, 275.7MB\n",
            "Optimizer stripped from runs/train/yolov5n-distilled4/weights/best.pt, 275.7MB\n",
            "\n",
            "Validating runs/train/yolov5n-distilled4/weights/best.pt...\n",
            "Fusing layers... \n",
            "Model Summary: 232 layers, 26887978 parameters, 0 gradients, 4.1 GFLOPs\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 9/9 [00:04<00:00,  1.89it/s]\n",
            "                 all        286        396      0.809      0.758      0.805      0.524\n",
            "                 cat        286         82      0.812      0.841      0.809      0.496\n",
            "                 cow        286         44      0.778      0.795      0.809      0.582\n",
            "               sheep        286         60      0.791      0.583      0.695      0.419\n",
            "                 dog        286        122      0.767      0.697      0.783       0.49\n",
            "               horse        286         88      0.895      0.874      0.928      0.632\n",
            "Results saved to \u001b[1mruns/train/yolov5n-distilled4\u001b[0m\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "!python val.py --data data/VOC_animal.yaml --weights runs/train/yolov5n-distilled4/weights/best.pt --batch-size 16"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "_NqO4rtF8yUc",
        "outputId": "8a2e6462-300c-49ea-f742-675e0897388e"
      },
      "execution_count": 10,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\u001b[34m\u001b[1mval: \u001b[0mdata=data/VOC_animal.yaml, weights=['runs/train/yolov5n-distilled4/weights/best.pt'], batch_size=16, imgsz=640, conf_thres=0.001, iou_thres=0.6, task=val, device=, workers=8, single_cls=False, augment=False, verbose=False, save_txt=False, save_hybrid=False, save_conf=False, save_json=False, project=runs/val, name=exp, exist_ok=False, half=False, dnn=False\n",
            "YOLOv5 🚀 3f3185b torch 1.12.1+cu113 CUDA:0 (Tesla T4, 15110MiB)\n",
            "\n",
            "Generating grammar tables from /usr/lib/python3.7/lib2to3/Grammar.txt\n",
            "Generating grammar tables from /usr/lib/python3.7/lib2to3/PatternGrammar.txt\n",
            "Fusing layers... \n",
            "Model Summary: 232 layers, 26887978 parameters, 0 gradients, 4.1 GFLOPs\n",
            "\u001b[34m\u001b[1mval: \u001b[0mScanning '/content/drive/.shortcut-targets-by-id/1-CXP81xST-t3P2hLZQLjfipgHlqh7k86/yolo-distillation/data/dataset/labels/val.cache' images and labels... 286 found, 0 missing, 7 empty, 0 corrupt: 100% 286/286 [00:00<?, ?it/s]\n",
            "               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100% 18/18 [00:04<00:00,  3.71it/s]\n",
            "                 all        286        396      0.838      0.732      0.807      0.525\n",
            "                 cat        286         82      0.818      0.768      0.806      0.496\n",
            "                 cow        286         44      0.771      0.766      0.822      0.592\n",
            "               sheep        286         60      0.853      0.583      0.696      0.416\n",
            "                 dog        286        122      0.815       0.68      0.781      0.488\n",
            "               horse        286         88      0.933      0.864       0.93      0.634\n",
            "Speed: 0.3ms pre-process, 3.3ms inference, 2.3ms NMS per image at shape (16, 3, 640, 640)\n",
            "Results saved to \u001b[1mruns/val/exp4\u001b[0m\n"
          ]
        }
      ]
    }
  ],
  "metadata": {
    "colab": {
      "collapsed_sections": [],
      "name": "Untitled0.ipynb",
      "provenance": []
    },
    "gpuClass": "standard",
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU"
  },
  "nbformat": 4,
  "nbformat_minor": 0
}